dgl.contrib.graph_store.SharedMemoryDGLGraph.init_ndata

SharedMemoryDGLGraph.init_ndata(ndata_name, shape, dtype, ctx=device(type='cpu'))[source]

Create node embedding.

It first creates the node embedding in the server and maps it to the current process with shared memory.

Parameters:
  • ndata_name (string) – The name of node embedding
  • shape (tuple) – The shape of the node embedding
  • dtype (string) – The data type of the node embedding. The currently supported data types are “float32” and “int32”.
  • ctx (DGLContext) – The column context.