We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f89731c commit b42c862Copy full SHA for b42c862
1 file changed
ppmat/models/newtonnet/scatter.py
@@ -1,5 +1,6 @@
1
from paddle_geometric.utils import scatter as scatter_org
2
3
+# The parameter may need to be converted to an integer before calling the original function, so add the function here
4
def scatter(src, index, dim=0, dim_size=None, reduce='sum'):
5
return scatter_org(src, index, dim=dim, dim_size=dim_size, reduce=reduce)
6
0 commit comments