debug

pyg-team · Jan 11, 2025 · 44a17f3 · 44a17f3
1 parent 644bcc3
commit 44a17f3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torch_geometric/data/large_graph_indexer.py b/torch_geometric/data/large_graph_indexer.py
@@ -539,9 +539,12 @@ def to_data(self, node_feature_name: str,
         node_id = torch.LongTensor(range(len(x)))
         print("self.get_edge_features(EDGE_INDEX)[:10]=",
               self.get_edge_features(EDGE_INDEX)[:10])
+        print("self.get_edge_features(EDGE_INDEX)[-10:]=",
+              self.get_edge_features(EDGE_INDEX)[-10:])
         edge_index = torch.t(
             torch.LongTensor(self.get_edge_features(EDGE_INDEX)))
-        print("edge_index tensor of full graph=", edge_index[:, :10])
+        print("edge_index tensor of full graph[:10]=", edge_index[:, :10])
+        print("edge_index tensor of full graph[-10:]=", edge_index[:, -10:])
 
         edge_attr = (self.get_edge_features(edge_feature_name)
                      if edge_feature_name is not None else None)