Функция PyTorch torch.std

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

torch.std— это функция в PyTorch, используемая для возврата стандартного отклонения тензора. Стандартное отклонение — это квадратный корень из дисперсии, измеряющий разброс данных.

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

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

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

Пример

import torch

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

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

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

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

标准差: tensor(1.5811)
dim=0 标准差: tensor([1.5000, 1.5000, 1.5000])
dim=1 标准差: tensor([1.0000, 1.0000])

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

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