PyTorch torch.var Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.varist eine Funktion in PyTorch, die verwendet wird, um die Varianz eines Tensors zurückzugeben. Die Varianz ist der Durchschnitt der quadrierten Differenzen zwischen jedem Datenpunkt und dem Mittelwert und misst die Streuung der Daten.

Funktionsdefinition

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

Anwendungsbeispiel

Beispiel

import torch

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

# Varianz aller Elemente zurückgeben
print("Varianz:", torch.var(x))

# Varianz entlang dim=0
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
print("Varianz für dim=0:", torch.var(y, dim=0))
print("Varianz für dim=1:", torch.var(y, dim=1))

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen