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