PyTorch torch.isin Funktion
PyTorch torch Referenzhandbuch
torch.isinEs ist eine Funktion in PyTorch, mit der geprüft wird, ob Elemente in einem anderen Tensor enthalten sind. Sie gibt einen booleschen Tensor zurück, der angibt, ob jedes Element des ersten Tensors im zweiten Tensor vorkommt.
Funktionsdefinition
torch.isin(elements, test_elements, assume_unique=False, invert=False)
Anwendungsbeispiele
Beispiel
import torch
# Zwei Tensoren erstellen
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Prüfen, ob Elemente in test_elements enthalten sind
result = torch.isin(elements, test_elements)
print(result)
# Zwei Tensoren erstellen
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Prüfen, ob Elemente in test_elements enthalten sind
result = torch.isin(elements, test_elements)
print(result)
Das Ausgabeergebnis ist:
tensor([False, True, False, True])
Andere Erweiterungen