torch.isin

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

测试elements中的每个元素是否在test_elements中。返回一个与elements形状相同的布尔张量，其中元素在test_elements中时为True，否则为False。

注意

其中一个 elements 或 test_elements 可以是标量，但不能同时是。

Parameters

• 元素 (Tensor 或 标量) – 输入元素

• test_elements (张量 或 标量) – 用于测试每个输入元素的值

• assume_unique (布尔值, 可选) – 如果为True，则假设elements和 test_elements都包含唯一元素，这可以加快计算速度。默认值：False

• invert (bool, 可选) – 如果为True，则反转布尔返回张量，对于不在test_elements中的元素，结果为True。默认值：False

Returns

与elements形状相同的布尔张量，其中在test_elements中的元素为True，否则为False

示例

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