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 或 純量) – 輸入元素

• test_elements (Tensor 或 純量) – 用於測試每個輸入元素的值

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