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 或 标量) – 用来与每个输入元素进行比较的测试值

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