search

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
21.28k stars 3.65k forks source link

Why does the DOCs GCN explanation differ to the GCN source code? #620

Closed EddyDavies closed 2 years ago

EddyDavies commented 5 years ago

Why is the GCN Docs code, which I think I understand, different from the GCN module source code?

rusty1s commented 5 years ago

The GCNConv doc code is a correct, but simplified implementation of the GCN operator. In addition, the GCNConv module in PyG implements

  1. linear transformations via own Parameter and matmul operations instead of using torch.nn.Linear
  2. caching behavior of the normalization for faster training in transductive scenarios
  3. the "improved" version from the Graph U-Net paper
  4. an optional bias