pytorch-geometric进阶
前向传播方式 [1]
- edge_index +
propagate
+message
: 这个主要是利用框架已有的方式进行处理。似乎主要是用torch_scatter封装了一下 - adj(sparse_tensor) +
propagate
+message_and_aggregate
: 主要是使用sparse_tensor,公式上写的好看一些。 - torch_scatter: 这个就不用message机制了,直接forward里一套带走,基本不用框架里的东西。[2]
- 这个好像只是对pytorch的接口包装了一下,改动应该不难。
数据集处理
主要是将图像上的算法迁移到graph上来所以需要这些
- 使用pytorch-geometric数据集可以直接通过index返回对应图的Data()结构[3]
- transoform和图像上的一样,都是作用于单个Data数据结构,可以通过这里的代码示例[4]看到
参考文献
Written on October 26, 2022