Fonction torch.var_mean de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.var_meanIl s'agit d'une fonction de PyTorch qui renvoie simultanément la variance et la moyenne d'un tenseur. Il s'agit d'une fonction efficace qui peut calculer la variance et la moyenne en une seule opération.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

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

# Renvoie simultanément la variance et la moyenne
var, mean = torch.var_mean(x)
print("Variance:", var)
print("Moyenne :", mean)

# Renvoie à la fois la variance et la moyenne selon 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("dim=0 variance:", var0)
print("moyenne pour dim=0 :", mean0)

La sortie est :

方差: 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 参考手册Manuel de référence Pytorch torch

Autres extensions