torch.isin

torch.isin(elements, test_elements, *, assume_unique=False, invert=False) → Tensor

測試 elements 中的每個元素是否包含在 test_elements 中。返回一個與 elements 形狀相同的布林張量，對於包含在 test_elements 中的元素，其值為 True，否則為 False。

注意

elements 或 test_elements 中的一個可以是標量，但不能兩者都是。

引數

• elements (Tensor 或 Scalar) – 輸入元素

• test_elements (Tensor 或 Scalar) – 用於測試每個輸入元素是否包含在其中的值

• assume_unique (bool, 可選) – 如果為 True，則假定 elements 和 test_elements 都包含唯一元素，這可以加快計算速度。預設值：False

• invert (bool, 可選) – 如果為 True，則反轉布林返回張量，對於**不**在 test_elements 中的元素，其值為 True。預設值：False

返回

一個與 elements 形狀相同的布林張量，對於包含在 test_elements 中的元素，其值為 True，否則為 False

示例

>>> torch.isin(torch.tensor([[1, 2], [3, 4]]), torch.tensor([2, 3]))
tensor([[False,  True],
        [ True, False]])