Fonction torch.var de PyTorch

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

torch.varest une fonction de PyTorch utilisée pour renvoyer la variance d'un tenseur. La variance est la moyenne des différences au carré entre chaque point de données et la moyenne, mesurant la dispersion des données.

Définition de fonction

torch.var(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 la variance de tous les éléments
print("Variance:", torch.var(x))

# Variance selon dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
print("dim=0 variance:", torch.var(y, dim=0))
print("dim=1 variance:", torch.var(y, dim=1))

Le résultat de sortie est:

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

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

Autres extensions