Среднее значение в PyTorch

Среднее арифметическое по тензору даёт метод mean: он делит сумму элементов на их количество. Для тензора из дробных чисел его можно вызвать сразу.

Если элементы целые, тензор сначала переводят к дробному типу методом float. Иначе среднее не вычислится. Создадим ряд оценок и выведем среднее:

import torch

scores = torch.tensor([2, 4, 6, 8])
average = scores.float().mean()
print(average) # выведет tensor(5.)

Та же цепочка работает и для таблицы, если нужно одно среднее по всем ячейкам:

import torch

table = torch.tensor([[1, 3], [5, 7]])
average = table.float().mean()
print(average) # выведет tensor(4.)

Соберите ряд цен 10, 20 и 30 и выведите среднее арифметическое одним числом.

Соберите таблицу баллов [[2, 4], [6, 8]] и выведите среднее по всем ячейкам.

Соберите ряд отметок 1, 2, 3 и 4 и выведите среднее одним числом.