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.