Если два узла имеют несколько ребер, как мне построить график с геометрическим pytorch?

Question

Предположим, что узел представляет собой банковский счет, а ребро представляет собой операцию перевода между двумя счетами. Между двумя аккаунтами может быть несколько переводов. если это так, как мне построить график с геометрическим pytorch?

Я строю график следующим образом:

      x = [[0], [1]] # 0 for account A and 1 for account B
edge_index = [ [0, 0, 0],
               [1,1,1]
             ]
edge_attr = [[...], [...], [...]]  # some edge attr such as transfer time, transfer amount
 
Data(x=x, edge_index=edge_index, edge_attr=edge_attr)

это верно?

Answer 1

Да, это работает. Однако это имеет смысл только тогда, когда типы данных для отдельных ребер идентичны, что, по-видимому, имеет место здесь, когда вы добавляете новое ребро для каждой транзакции.