Función PyTorch torch.var_mean

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

torch.var_meanEs una función en PyTorch que devuelve simultáneamente la varianza y la media de un tensor. Es una función eficiente que puede calcular la varianza y la media al mismo tiempo en una sola operación.

Definición de la función

torch.var_mean(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 y la media al mismo tiempo
var, mean = torch.var_mean(x)
print("Varianza:", var)
print("Media:", mean)

# Devuelve la varianza y la media a lo largo de dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
var0, mean0 = torch.var_mean(y, dim=0)
print("Varianza en dim=0:", var0)
print("Media en dim=0:", mean0)

El resultado es:

方差: tensor(2.5000)
均值: tensor(3.)
dim=0 方差: tensor([2.2500, 2.2500, 2.2500])
dim=0 均值: tensor([2.5000, 3.5000, 4.5000])

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

Otras extensiones