Я хочу получить список смежности с помощью pytorch geo. Есть ли какой-нибудь метод в pytorch для преобразования матрицы ⇐ Python

[Anonymous]

Я хочу получить список смежности с помощью pytorch geo. Есть ли в pytorch какой-нибудь метод для этого или что-то еще для преобразования матрицы смежности, но оптимизированной, а не циклической?
Вот так:
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


Подробнее здесь: https://stackoverflow.com/questions/784 ... pytorch-do