Jumping Knowledge Setting Last vs None

[Bobowner]

📚 Describe the documentation issue

It is unclear what the difference between None and last for the parameter jk is: https://pytorch-geometric.readthedocs.io/en/latest/generated/torch_geometric.nn.models.GCN.html

since the JK Paper only describes cat, lstm, and max.

Suggest a potential alternative/fix

No response

[rusty1s]

Sorry for the confusion. If jk Is specified, the model will additionally apply a final linear transformation to transform node embeddings to the expected output feature dimensionality. As such, jk="last" will perform a final linear transformation, while jk=None will not.