torch_geometric.utils的to_dense_adj和to_dense_batch函数解读

1,955 阅读1分钟
  • 笔记人:陈亦新

最近遇到了两个贼难理解的函数:

  • torch_geometric.utils.to_dense_batch
  • torch_geometric.utils.to_dense_adj

to_dense_adj

image.png

  • 将batched稀疏拼接矩阵转换成单个稠密邻接矩阵

to_dense_bath

image.png

例子1:

image.png

例子2:

image.png