torch.isin
torch.isin(elements, test_elements, *, assume_unique=False, invert=False) -> Tensor
Tests if each element of "elements" is in "test_elements". Returns
   a boolean tensor of the same shape as "elements" that is True for
   elements in "test_elements" and False otherwise.
Note:
 One of "elements" or "test_elements" can be a scalar, but not
 both.

Parameters:
      * elements (Tensor or Scalar) -- Input elements
  * **test_elements** (*Tensor** or **Scalar*) -- Values against
    which to test for each input element

  * **assume_unique** (*bool**, **optional*) -- If True, assumes
    both "elements" and "test_elements" contain unique elements,
    which can speed up the calculation. Default: False

  * **invert** (*bool**, **optional*) -- If True, inverts the
    boolean return tensor, resulting in True values for elements
    *not* in "test_elements". Default: False

Returns: