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, optional) – 如果為 True，則假定 elements 和 test_elements 都包含唯一元素，這可以加快計算速度。預設為：False
invert (bool, optional) – 如果為 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]])

文件