PyTorch torch.var función

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

torch.vares una función en PyTorch que se utiliza para devolver la varianza de un tensor. La varianza es el promedio de los cuadrados de las diferencias entre cada dato y la media, y mide el grado de dispersión de los datos.

Definición de la función

torch.var(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 varianza de todos los elementos
print("Varianza:", torch.var(x))

# Varianza a lo largo de dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
print("Varianza de dim=0:", torch.var(y, dim=0))
print("Varianza de dim=1:", torch.var(y, dim=1))

El resultado de salida es:

方差: tensor(2.5000)
dim=0 方差: tensor([2.2500, 2.2500, 2.2500])
dim=1 方差: tensor([1., 1.])

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

Otras extensiones