dgl.NodeFlow.copy_from_parent

NodeFlow.copy_from_parent(node_embed_names='__ALL__', edge_embed_names='__ALL__', ctx=None)[source]

Copy node/edge features from the parent graph.

Parameters

  • node_embed_names (a list of lists of strings, optional) – The names of embeddings in each layer.

  • edge_embed_names (a list of lists of strings, optional) – The names of embeddings in each block.

  • ctx (Context) – The device to copy tensor to. If None, features will stay at its original device