Я хочу получить список смежности с помощью 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
