
Графовые нейронные сети: выявление зависимостей в структурированных графовых данных
625
# пакет пр едставляет со бой список слов арей, каждый из которых
# содержит представлени е и метку графа
def collate_graphs(batch):
adj_mats = [graph['A'] for graph in batch]
sizes = [A.size(0) for A in adj_mats]
tot_size = sum(sizes)
# созд аем пакетную матрицу
batch_mat = get_batch_tensor(sizes)
# объе диняем матриц ы признаков
feat_mats = torch.cat([graph['X'] for graph in batch], dim=0)
# объе диняем метки
labels = torch.cat([graph['y'] for graph in batch], dim=0)
# ообъ единяем матри цы смежности
batch_adj = torch.zeros([tot_size, tot_size], ...