PyTorch torch.isin Function
Pytorch torch Reference Manual
torch.isinIt is a function in PyTorch used to check whether elements are in another tensor. It returns a boolean tensor indicating whether each element of the first tensor appears in the second tensor.
Function Definition
torch.isin(elements, test_elements, assume_unique=False, invert=False)
Usage Examples
Example
import torch
# Create two tensors
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Check whether elements are in test_elements
result = torch.isin(elements, test_elements)
print(result)
# Create two tensors
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Check whether elements are in test_elements
result = torch.isin(elements, test_elements)
print(result)
The output result is:
tensor([False, True, False, True])
Other Extensions