rllm.utils.to_undirected

class rllm.utils.to_undirected(adj: Tensor)

Bases:

Convert the given adjacency matrix to an undirected graph representation by adding reverse edges.

Parameters:

adj (Tensor) – The adjacency matrix in sparse COO format.

Returns:

The symmetrized (undirected) adjacency matrix in sparse COO format.

Return type:

Tensor