dgl.NodeFlow.apply_layer

NodeFlow.apply_layer(layer_id, func='default', v='__ALL__', inplace=False)[source]

Apply node update function on the node embeddings in the specified layer.

Parameters

  • layer_id (int) – The specified layer to update node embeddings.

  • func (callable or None, optional) – Apply function on the nodes. The function should be a Node UDF.

  • v (a list of vertex Ids or ALL.) – The vertex Ids (unique in the NodeFlow) to run the node update function.

  • inplace (bool, optional) – If True, update will be done in place, but autograd will break.