PyTorch torch.all Funktion
PyTorch torch Referenzhandbuch
torch.allEs ist eine Funktion in PyTorch, mit der festgestellt wird, ob alle Elemente True sind.
Funktionsdefinition
torch.all(input, dim, keepdim, out)
Anwendungsbeispiele
Beispiel
import torch
x = torch.tensor([[True, True, True], [True, False, True]])
# Globale Prüfung
print("Alle sind True:", torch.all(x))
# Überprüfung entlang der Dimension
print("dim=0 alle sind True:", torch.all(x, dim=0))
x = torch.tensor([[True, True, True], [True, False, True]])
# Globale Prüfung
print("Alle sind True:", torch.all(x))
# Überprüfung entlang der Dimension
print("dim=0 alle sind True:", torch.all(x, dim=0))
Die Ausgabe ist:
全部为 True: tensor(False) dim=0 全部为 True: tensor([True, False, True])
Andere Erweiterungen