PyTorch torch.mean Funktion
PyTorch torch Referenzhandbuch
torch.meanEs ist eine Funktion in PyTorch, die verwendet wird, um den Mittelwert eines Tensors zu berechnen.
Funktionsdefinition
torch.mean(input, dim, keepdim, dtype, out)
Anwendungsbeispiele
Beispiel
import torch
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Mittelwert eines Tensors berechnen
print("Mittelwert:", torch.mean(x))
# Mittelwert entlang dim=0 berechnen
print("dim=0 Mittelwert:", torch.mean(x, dim=0))
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Mittelwert eines Tensors berechnen
print("Mittelwert:", torch.mean(x))
# Mittelwert entlang dim=0 berechnen
print("dim=0 Mittelwert:", torch.mean(x, dim=0))
Das Ausgabeergebnis ist:
全局均值: tensor(3.5000) dim=0 均值: tensor([2.5000, 3.5000, 4.5000])
Andere Erweiterungen