Fonction PyTorch torch.isin
Manuel de référence Pytorch torch
torch.isinC'est une fonction de PyTorch utilisée pour vérifier si des éléments se trouvent dans un autre tenseur. Elle renvoie un tenseur booléen indiquant si chaque élément du premier tenseur apparaît dans le second tenseur.
Définition de fonction
torch.isin(elements, test_elements, assume_unique=False, invert=False)
Exemples d'utilisation
Exemple
import torch
# Créer deux tenseurs
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Vérifier si les éléments sont dans test_elements
result = torch.isin(elements, test_elements)
print(result)
# Créer deux tenseurs
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Vérifier si les éléments sont dans test_elements
result = torch.isin(elements, test_elements)
print(result)
Le résultat de sortie est:
tensor([False, True, False, True])
Autres extensions