【pytorch】forループで作ったtensorを空のリストを作らずに結合

解決策

グラフ特徴量の計算を行う関数の一部

for i in range(num_nodes):
    ############省略############
    edge_idx, edge_att = coalesce(edge_idx, edge_att, num_nodes, num_nodes)
    try:
        edge_idx_list = torch.stack((edge_idx_list, edge_idx), dim = 0)
    except:
        edge_idx_list = edge_idx

個人的なメモ:問題

GNNの入力のためにエッジのインデックスとエッジ特徴量の配列をグラフごとに計算し、datasetにぶち込むためにそれぞれを一つのtensorに収めようとした。
計算後のtensorを空のリストにappendして最後にtorchに変換する方法を試みたが、リスト内の要素の型がtorch.tensorになっていたりリストからtorch.tensorに変換しようとするのにうまい方法が分からなかった

参照

• How to append to a tensor in a loop