[FIXED] How to check whether tensor values in a different tensor pytorch?

Issue

I have 2 tensors of unequal size

a = torch.tensor([[1,2], [2,3],[3,4]])
b = torch.tensor([[4,5],[2,3]])

I want a boolean array of whether each value exists in the other tensor without iterating. something like

a in b

and the result should be

[False, True, False]

as only the value of a[1] is in b

Solution

I think it's impossible without using at least some type of iteration. The most succinct way I can manage is using list comprehension:

[True if i in b else False for i in a]

Checks for elements in b that are in a and gives [False, True, False]. Can also be reversed to get elements a in b [False, True].

Answered By - AverageHomosapien