快捷方式

torch.isin

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

测试 elements 的每个元素是否在 test_elements 中。返回与 elements 形状相同的布尔张量,对于在 test_elements 中的元素为 True,否则为 False。

注意

elementstest_elements 之一可以是标量,但不能两者都是。

参数
  • elements (TensorScalar) – 输入元素

  • test_elements (TensorScalar) – 对每个输入元素进行测试的值

  • assume_unique (bool, 可选) – 如果为 True,则假定 elementstest_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]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并解答疑问

查看资源