Fonction torch.std_mean de PyTorch

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

torch.std_meanest une fonction dans PyTorch qui renvoie simultanément l'écart-type et la moyenne d'un tenseur. C'est une fonction efficace qui peut calculer l'écart-type et la moyenne en une seule opération.

Définition de fonction

torch.std_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])

# Renvoyer l'écart-type et la moyenne simultanément
std, mean = torch.std_mean(x)
print("Écart-type :", std)
print("Moyenne :", mean)

# Renvoie l'écart-type et la moyenne simultanément selon dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
std0, mean0 = torch.std_mean(y, dim=0)
print("écart-type pour dim=0 :", std0)
print("moyenne pour dim=0 :", mean0)

Le résultat de sortie est:

标准差: tensor(1.5811)
均值: tensor(3.)
dim=0 标准差: tensor([1.5000, 1.5000, 1.5000])
dim=0 均值: tensor([2.5000, 3.5000, 4.5000])

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

Autres extensions