Как построить NN с входным : 1D вектором размера n и выходным 2D вектором с размером n*n

#python #tensorflow #graph #neural-network #adjacency-matrix

Вопрос:

Я хочу обучить NN для создания матрицы смежности из списка функций узлов, поэтому моя сеть должна быть LSTM или CNN,

вход:

  • массив n*nb_features ( узлов ) ; например, узел представляет собой шар и может быть представлен следующими функциями [x1,x2,x3,r] с (x1,x2,x3) евклидовыми координатами центра и r радиусом

выход:

  • первый случай: двумерный массив с n*n, представляющий список смежности, явно,элемент i, j имеет значение 1, если шар i и шар j являются смежными геометрически
  • второй случай : список смежности, представляющий собой список всех пар (i,j), таких, что узел (шар) i и узел j являются смежными