ここ最近Transformerの話題が多いですがなかなか理解の大変なモジュールなので、下記の2章の記述を公開することにいたしました。
https://lib-arts.booth.pm/items/2741653
処理の流れだけを追ってもわかりづらい可能性があるため、Message Passingに基づくGraph Neural Networkとして理解するのが良いかと思います。
Deep Graph Libraryのチュートリアルの記載からの抜粋が多いので、詳しく考えたい方はそちらも参照してみてください。
※
引用にあたっての記載があまりよくないようなので、近いうちに修正します。