dgl.graphbolt.isin

dgl.graphbolt.isin(elements, test_elements)

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

Parameters:

• 元素 (torch.Tensor) – 一个一维张量，表示输入元素。

• test_elements (torch.Tensor) – 一个一维张量，表示每个输入要测试的值。

示例

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