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.