We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a1df0a2 commit 5e07dd3Copy full SHA for 5e07dd3
torchdrug/data/graph.py
@@ -334,7 +334,7 @@ def split(self, node2graph):
334
edge_index = edge2graph.argsort()
335
edge_index = edge_index[edge_mask[edge_index]]
336
337
- prepend = -torch.tensor([-1], device=self.device)
+ prepend = torch.tensor([-1], device=self.device)
338
is_first_node = torch.diff(node2graph[index], prepend=prepend) > 0
339
graph_index = self.node2graph[index[is_first_node]]
340
0 commit comments