Python _pop_subgraph_storageの例

コード例 #1

ファイル: dataloading.py プロジェクト: wangxiaoyunNV/dgl

 def collator(self, items):
     result = super().collate(items)
     # Copy the feature from parent graph
     _pop_subgraph_storage(result[1], self.g)
     _pop_subgraph_storage(result[2], self.g)
     _pop_storages(result[-1], self.g_sampling)
     return result

コード例 #2

ファイル: dataloading.py プロジェクト: wangxiaoyunNV/dgl

 def collator(self, items):
     """
     The interface of collator, input items is edge id of the attached graph
     """
     result = super().collate(items)
     # Copy the feature from parent graph
     _pop_subgraph_storage(result[1], self.g)
     _pop_subgraph_storage(result[2], self.g)
     _pop_storages(result[-1], self.g_sampling)
     return result