How can I get the edge_index tensor from the adjancency matrix?

[codyseally]

My data object has the data.adj_t parameter, giving me the sparse adjacency matrix. How can I get the edge_index tensor of size [2, num_edges] from this?

[rusty1s]

row, col, _ = data.adj_t.().coo()
edge_index = torch.stack([row, col], dim=0)