rllm.utils.get_indices¶

class rllm.utils.get_indices(adj: Tensor)[source]¶

Bases:

Get indices of non-zero elements from an adjacency matrix.

Parameters:: adj (Tensor) – the adjacency matrix.
Returns:: indices of non-zero elements.
Return type:: indices (Tensor)