torch.isin¶

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

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

注意

一个可以是标量，但不能同时是两个。

参数:

元素（张量或标量）- 输入元素
test_elements（张量或标量）- 用于测试每个输入元素的值
assume_unique（布尔值，可选）- 如果为 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]])

文档