Функция PyTorch torch.var_mean

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.var_meanЭто функция в PyTorch, которая одновременно возвращает дисперсию и среднее тензора. Это эффективная функция, которая может вычислить дисперсию и среднее за одну операцию.

Определение функции

torch.var_mean(input, dim, unbiased, keepdim=False)

Пример использования

Пример

import torch

x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])

# Возвращает одновременно дисперсию и среднее
var, mean = torch.var_mean(x)
print("Дисперсия:", var)
print("Среднее:", mean)

# Возвращает и дисперсию, и среднее по dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
var0, mean0 = torch.var_mean(y, dim=0)
print("дисперсия при dim=0:", var0)
print("Среднее при dim=0:", mean0)

Вывод:

方差: tensor(2.5000)
均值: tensor(3.)
dim=0 方差: tensor([2.2500, 2.2500, 2.2500])
dim=0 均值: tensor([2.5000, 3.5000, 4.5000])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения