Функция std
Функция std вычисляет стандартное отклонение элементов тензора.
Первым параметром функция принимает тензор.
Вторым параметром можно передать размерность dim, вдоль которой выполняется вычисление.
Третьим параметром можно передать логическое значение unbiased для выбора формулы вычисления (смещенная или несмещенная оценка).
Четвертым параметром можно передать keepdim для сохранения размерности.
По умолчанию функция вычисляет стандартное отклонение по всем элементам тензора.
Синтаксис
torch.std(input)
torch.std(input, dim)
torch.std(input, dim, unbiased)
torch.std(input, dim, unbiased, keepdim)
Пример
Давайте вычислим стандартное отклонение всех элементов тензора чисел:
import torch
t = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
res = torch.std(t)
print(res)
Результат выполнения кода:
tensor(1.5811)
Пример
Давайте вычислим стандартное отклонение вдоль строк двумерного тензора с сохранением размерности:
import torch
t = torch.tensor([
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0],
])
res = torch.std(t, dim=0, keepdim=True)
print(res)
Результат выполнения кода:
tensor([
[2.1213, 2.1213, 2.1213],
])
Пример
Давайте вычислим стандартное отклонение с использованием смещенной оценки (unbiased=False):
import torch
t = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
res = torch.std(t, unbiased=False)
print(res)
Результат выполнения кода:
tensor(1.4142)
Пример
Давайте вычислим стандартное отклонение вдоль столбцов двумерного тензора:
import torch
t = torch.tensor([
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0],
])
res = torch.std(t, dim=1)
print(res)
Результат выполнения кода:
tensor([1.0000, 1.0000])
