i want to get adjacency list with pytorch geo, there is any methode in pytorch do that or anything to convert adjacency matrix but optimized, not loop

like this :
import numpy as np
import networkx as nx

def adjacency_matrix_to_list(adj_matrix):
# Créer un graphe à partir de la matrice d’adjacence
graph = nx.from_numpy_matrix(adj_matrix)

# Convertir le graphe en liste d'adjacence
adj_list = nx.to_dict_of_lists(graph)

return adj_list