Функция torch.std_mean в PyTorch

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

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

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

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

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

Пример

import torch

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

# Возвращает стандартное отклонение и среднее одновременно
std, mean = torch.std_mean(x)
print("Стандартное отклонение:", std)
print("Среднее:", mean)

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

Результат вывода:

标准差: tensor(1.5811)
均值: tensor(3.)
dim=0 标准差: tensor([1.5000, 1.5000, 1.5000])
dim=0 均值: tensor([2.5000, 3.5000, 4.5000])

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

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