Функция mean
Функция mean вычисляет среднее арифметическое значение элементов тензора.
Первым параметром функция принимает тензор. Вторым параметром можно передать
измерение (ось), вдоль которого будет производиться вычисление.
Третьим параметром можно указать, нужно ли сохранять размерность.
Возвращает функция тензор со средними значениями.
Синтаксис
torch.mean(input, [dim], [keepdim])
Пример
Давайте вычислим среднее арифметическое всех элементов тензора:
import torch
t = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
res = torch.mean(t)
print(res)
Результат выполнения кода:
tensor(3.)
Пример
Давайте вычислим среднее значение по строкам двумерного тензора (измерение 0):
import torch
t = torch.tensor([
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0],
])
res = torch.mean(t, dim=0)
print(res)
Результат выполнения кода:
tensor([2.5000, 3.5000, 4.5000])
Пример
Давайте вычислим среднее значение по столбцам (измерение 1) с сохранением размерности:
import torch
t = torch.tensor([
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0],
])
res = torch.mean(t, dim=1, keepdim=True)
print(res)
Результат выполнения кода:
tensor([
[2.],
[5.],
])
Пример
Давайте используем метод тензора для вычисления среднего:
import torch
t = torch.tensor([10.0, 20.0, 30.0])
res = t.mean()
print(res)
Результат выполнения кода:
tensor(20.)
