PyTorch torch.amax Funktion
PyTorch torch-Referenzhandbuch
torch.amaxEs ist eine Funktion in PyTorch, die verwendet wird, um den Maximalwert eines Tensors entlang der angegebenen Dimension zurückzugeben.
Funktionsdefinition
torch.amax(input, dim, keepdim=False)
Anwendungsbeispiele
Beispiel
import torch
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
# Den Maximalwert aller Elemente zurückgeben
print("Globales Maximum:", torch.amax(x))
# Maximum entlang dim=0
print("Maximum für dim=0:", torch.amax(x, dim=0))
# Maximum entlang dim=1
print("Maximum für dim=1:", torch.amax(x, dim=1))
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
# Den Maximalwert aller Elemente zurückgeben
print("Globales Maximum:", torch.amax(x))
# Maximum entlang dim=0
print("Maximum für dim=0:", torch.amax(x, dim=0))
# Maximum entlang dim=1
print("Maximum für dim=1:", torch.amax(x, dim=1))
Das Ausgabeergebnis ist:
全局最大: tensor(4) dim=0 最大: tensor([4, 3, 3]) dim=1 最大: tensor([3, 4])
Andere Erweiterungen