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]])