PyTorch torch.any Funktion
PyTorch torch Referenzhandbuch
torch.anyEs ist eine Funktion in PyTorch, die verwendet wird, um festzustellen, ob irgendein Element True ist.
Funktionsdefinition
torch.any(input, dim, keepdim, out)
Anwendungsbeispiel
Beispiel
import torch
x = torch.tensor([[True, False, False], [False, False, False]])
print("Enthält True:", torch.any(x))
print("dim=0 Has True:", torch.any(x, dim=0))
x = torch.tensor([[True, False, False], [False, False, False]])
print("Enthält True:", torch.any(x))
print("dim=0 Has True:", torch.any(x, dim=0))
Das Ausgabeergebnis ist:
有 True: tensor(True) dim=0 有 True: tensor([True, False, False])
Andere Erweiterungen