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
形状相同的布尔张量,对于在test_elements
中的元素为 True,否则为 False
示例
>>> torch.isin(torch.tensor([[1, 2], [3, 4]]), torch.tensor([2, 3])) tensor([[False, True], [ True, False]])