天天看點

torch_geometric.utils的to_dense_adj和to_dense_batch函數解讀

  • 筆記人:陳亦新

最近遇到了兩個賊難了解的函數:

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

to_dense_adj

  • 将batched稀疏拼接矩陣轉換成單個稠密鄰接矩陣

to_dense_bath

例子1:

繼續閱讀