-
Notifications
You must be signed in to change notification settings - Fork 3.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Tutorial on nn.aggr
package
#5191
Conversation
Codecov Report
@@ Coverage Diff @@
## master #5191 +/- ##
=======================================
Coverage 83.05% 83.05%
=======================================
Files 335 335
Lines 18485 18485
=======================================
Hits 15353 15353
Misses 3132 3132
📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Some very minor nits, but overall looks great! LGTM
docs/source/modules/nn.rst
Outdated
|
||
output = mean_aggr(x, index) # Output shape: [100, 64] | ||
|
||
Notably, all aggregations share the same set of forward arguments, as described in detail in :meth:`torch_geometric.nn.aggr.Aggregation.forward`. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Do we want to caveat here that there is not yet support for these arguments across all aggregations?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Good point. I think we should do this in the doc-strings of the individual modules, using a warning box for example. Let me know if you want to add this in.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice :)
Co-authored-by: Padarn Wilson <padarn.wilson@grabtaxi.com>
Co-authored-by: Manan Shah <manan.shah.777@gmail.com>
No description provided.