dgl.NodeFlow.block_adjacency_matrix

NodeFlow.block_adjacency_matrix(block_id, ctx)[source]

Return the adjacency matrix representation for a specific block in a NodeFlow.

A row of the returned adjacency matrix represents the destination of an edge and the column represents the source.

Parameters:
  • block_id (int) – The specified block to return the adjacency matrix.
  • ctx (context) – The context of the returned matrix.
Returns:

  • SparseTensor – The adjacency matrix.
  • Tensor – A index for data shuffling due to sparse format change. Return None if shuffle is not required.