Ejemplos de _pop_subgraph_storage en Python

Ejemplo n.º 1

Archivo: dataloading.py Proyecto: 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

Ejemplo n.º 2

Archivo: dataloading.py Proyecto: 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