Get DGLGraph from GraphData
(self)
| 67 | return _CAPI_MakeGraphData(ghandle, node_tensors, edge_tensors) |
| 68 | |
| 69 | def get_graph(self): |
| 70 | """Get DGLGraph from GraphData""" |
| 71 | ghandle = _CAPI_GDataGraphHandle(self) |
| 72 | hgi = _CAPI_DGLAsHeteroGraph(ghandle) |
| 73 | g = DGLGraph(hgi, ["_U"], ["_E"]) |
| 74 | node_tensors_items = _CAPI_GDataNodeTensors(self).items() |
| 75 | edge_tensors_items = _CAPI_GDataEdgeTensors(self).items() |
| 76 | for k, v in node_tensors_items: |
| 77 | g.ndata[k] = F.zerocopy_from_dgl_ndarray(v) |
| 78 | for k, v in edge_tensors_items: |
| 79 | g.edata[k] = F.zerocopy_from_dgl_ndarray(v) |
| 80 | return g |
| 81 | |
| 82 | |
| 83 | def save_graphs(filename, g_list, labels=None, formats=None): |
no test coverage detected