import torch A = torch.tensor([1,2,3,4]) b = torch.tensor([2]) ind_for_b_in_A = torch.nonzero(torch.eq(A, b)).squeeze(1)