PyTorch torch.max Funktion
PyTorch torch Referenzhandbuch
torch.maxSie ist eine Funktion in PyTorch, die verwendet wird, um den Maximalwert von Tensoren zu berechnen.
Funktionsdefinition
torch.max(input, dim, keepdim, out)
Anwendungsbeispiele
Beispiel
import torch
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Globales Maximum
print("Globales Maximum:", torch.max(x))
# Maximum entlang der Dimension
print("Maximum entlang dim=0:", torch.max(x, dim=0))
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Globales Maximum
print("Globales Maximum:", torch.max(x))
# Maximum entlang der Dimension
print("Maximum entlang dim=0:", torch.max(x, dim=0))
Das Ausgabeergebnis ist:
全局最大: tensor(6) dim=0 最大: torch.return_types.max(values=tensor([4, 5, 6]), indices=tensor([1, 1, 1]))
Andere Erweiterungen