dgl.graphbolt.isin

dgl.graphbolt.isin(elements, test_elements)

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

参数:

• elements (torch.Tensor) – 表示输入元素的 1D 张量。

• test_elements (torch.Tensor) – 表示用于对每个输入进行测试的值的 1D 张量。

示例

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