dgl.NodeFlow.apply_layer

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

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.