Función PyTorch torch.mean
Manual de referencia de Pytorch torch
torch.meanEs una función en PyTorch que se utiliza para calcular la media de tensores.
Definición de la función
torch.mean(input, dim, keepdim, dtype, out)
Ejemplo de uso
Ejemplo
import torch
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Calcula la media de todos los elementos
print("Media global:", torch.mean(x))
# Calcula la media a lo largo de dim=0
print("Media dim=0:", torch.mean(x, dim=0))
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Calcula la media de todos los elementos
print("Media global:", torch.mean(x))
# Calcula la media a lo largo de dim=0
print("Media dim=0:", torch.mean(x, dim=0))
El resultado de salida es:
全局均值: tensor(3.5000) dim=0 均值: tensor([2.5000, 3.5000, 4.5000])
Otras extensiones