dgl.NodeFlow.apply_block

NodeFlow.apply_block(block_id, func='default', edges='__ALL__', inplace=False)[source]

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

Parameters:
  • block_id (int) – The specified block to update edge embeddings.
  • func (callable or None, optional) – Apply function on the edges. The function should be an Edge UDF.
  • edges (a list of edge Ids or ALL.) – The edges Id to run the edge update function.
  • inplace (bool, optional) – If True, update will be done in place, but autograd will break.