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.