PyTorch torch.std función

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.stdEs una función en PyTorch que se utiliza para devolver la desviación estándar de un tensor. La desviación estándar es la raíz cuadrada de la varianza y mide la dispersión de los datos.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

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

# Devuelve la desviación estándar de todos los elementos
print("Desviación estándar:", torch.std(x))

# Desviación estándar a lo largo de dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
print("Desviación estándar dim=0:", torch.std(y, dim=0))
print("Desviación estándar dim=1:", torch.std(y, dim=1))

El resultado es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones