[Add] gather_nd scatter_nd (#5422)
* add gather_nd and scatter_nd * add testcase * stable_api * add docs and doctest * add scatter_nd grad * gradfunc * delete some code comments * add grad test * add docs * update words usage * delete some import * update * update * auto format by CI Co-authored-by:Yao Chi <later@usopp.net> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org>
Showing
- docs/source/experimental.rst 2 additions, 0 deletionsdocs/source/experimental.rst
- oneflow/core/autograd/gradient_funcs/scatter_nd.cpp 56 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/scatter_nd.cpp
- oneflow/python/nn/modules/gather_nd.py 84 additions, 0 deletionsoneflow/python/nn/modules/gather_nd.py
- oneflow/python/nn/modules/scatter_nd.py 73 additions, 0 deletionsoneflow/python/nn/modules/scatter_nd.py
- oneflow/python/ops/array_ops.py 2 additions, 0 deletionsoneflow/python/ops/array_ops.py
- oneflow/python/test/modules/test_gather_nd.py 104 additions, 0 deletionsoneflow/python/test/modules/test_gather_nd.py
- oneflow/python/test/modules/test_scatter_nd.py 105 additions, 0 deletionsoneflow/python/test/modules/test_scatter_nd.py
Please register or sign in to comment