PyTorch torch.isin función
Pytorch torch manual de referencia
torch.isinEs una función en PyTorch que se utiliza para comprobar si los elementos están en otro tensor. Devuelve un tensor booleano que indica si cada elemento del primer tensor aparece en el segundo tensor.
Definición de la función
torch.isin(elements, test_elements, assume_unique=False, invert=False)
Ejemplo de uso
Ejemplo
import torch
# Crear dos tensores
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Comprobar si los elementos están en test_elements
result = torch.isin(elements, test_elements)
print(result)
# Crear dos tensores
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Comprobar si los elementos están en test_elements
result = torch.isin(elements, test_elements)
print(result)
El resultado de salida es:
tensor([False, True, False, True])
Otras extensiones