PyTorch torch.std Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.stdist eine Funktion in PyTorch, die verwendet wird, um die Standardabweichung eines Tensors zurückzugeben. Die Standardabweichung ist die Quadratwurzel der Varianz und misst die Streuung der Daten.

Funktionsdefinition

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

Anwendungsbeispiel

Beispiel

import torch

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

# Gibt die Standardabweichung aller Elemente zurück
print("Standardabweichung:", torch.std(x))

# Standardabweichung entlang dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
print("dim=0 Standardabweichung:", torch.std(y, dim=0))
print("dim=1 Standardabweichung:", torch.std(y, dim=1))

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen