Given tensor A=torch.tensor([[2, 1, 5], [3, 2, 4]]) and B=torch.tensor([1, 3, 5])
How can I return tensor C that tells whether element in A is in B, i.e. C=torch.tensor([[False, True, True], [True, False, False]]).
2 Likes
You can do this with a for loop:
>>> sum(A==i for i in B).bool()
tensor([[False, True, True],
[ True, False, False]])
1 Like