Fonction torch.std de PyTorch

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

torch.stdest une fonction dans PyTorch utilisée pour renvoyer l'écart-type d'un tenseur. L'écart-type est la racine carrée de la variance et mesure la dispersion des données.

Définition de fonction

torch.std(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 de tous les éléments
print("Écart-type :", torch.std(x))

# Écart-type selon dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
print("écart-type pour dim=0 :", torch.std(y, dim=0))
print("écart-type pour dim=1 :", torch.std(y, dim=1))

Le résultat de sortie est:

标准差: tensor(1.5811)
dim=0 标准差: tensor([1.5000, 1.5000, 1.5000])
dim=1 标准差: tensor([1.0000, 1.0000])

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

Autres extensions