[GitHub] sxjscience opened a new pull request #9200: Fix the gradient of gather_nd

[GitBox <gi...@apache.org>]

sxjscience opened a new pull request #9200: Fix the gradient of gather_nd
URL: https://github.com/apache/incubator-mxnet/pull/9200
 
 
   ## Description ##
   Add scatter_nd_acc, which accumulates the value when the indices are same. `gather_nd` and `scatter_nd_acc` are reverse to each other.
   
   ## Checklist ##
   - [x] Passed code style checking (`make lint`)
   - [x] Changes are complete (i.e. I finished coding on this PR)
   - [x] All changes have test coverage:
   - [x] Code is well-documented: 
   - [x] To the my best knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change
   
   ### Changes ###
   - [x] Add new type switch macro that can be used when int8 is not supported
   - [x] atomicAdd support for int64
   - [x] Add scatter_nd_acc
   - [x] Set the gradient of gather_nd to scatter_nd_acc
   
   ## Comments ##
   I use atomicAdd to implement the operator. The current CPU implementation does not used openmp. Also, int8 and uint8 are not supported.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services