Функция torch.mean в PyTorch
Справочное руководство Pytorch torch
torch.meanЭто функция в PyTorch, используемая для вычисления среднего значения тензора.
Определение функции
torch.mean(input, dim, keepdim, dtype, out)
Примеры использования
Пример
import torch
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Вычислить среднее всех элементов
print("Глобальное среднее:", torch.mean(x))
# Вычислить среднее по dim=0
print("среднее по dim=0:", torch.mean(x, dim=0))
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Вычислить среднее всех элементов
print("Глобальное среднее:", torch.mean(x))
# Вычислить среднее по dim=0
print("среднее по dim=0:", torch.mean(x, dim=0))
Результат вывода:
全局均值: tensor(3.5000) dim=0 均值: tensor([2.5000, 3.5000, 4.5000])
Другие расширения