Python PyTorch isin用法及代碼示例

本文簡要介紹python語言中 torch.isin 的用法。

用法:

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

參數：

• elements(Tensor或者標量) -輸入元素

• test_elements(Tensor或者標量) -測試每個輸入元素的值

• assume_unique(bool,可選的) -如果為 True，則假定 elements 和 test_elements 都包含唯一元素，這可以加快計算速度。默認值：假

• invert(bool,可選的) -如果為 True，則反轉布爾返回張量，從而為元素生成 True 值不是在test_elements。默認值：假

返回：

與 elements 形狀相同的布爾張量，對於 test_elements 中的元素為 True，否則為 False

測試 elements 的每個元素是否在 test_elements 中。返回與 elements 形狀相同的布爾張量，對於 test_elements 中的元素為 True，否則為 False。

注意

elements 或 test_elements 之一可以是標量，但不能同時是標量。

示例

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