Add gather embedding module (#4826)
* add gather module * add gather module * add test case * add embedding module * fix comments * update embedding module and test case * refine * fix comment * fix comment * fix comment Co-authored-by:Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Co-authored-by:
BBuf <1182563586@qq.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- docs/source/experimental.rst 4 additions, 0 deletionsdocs/source/experimental.rst
- oneflow/python/nn/modules/gather.py 98 additions, 0 deletionsoneflow/python/nn/modules/gather.py
- oneflow/python/nn/modules/sparse.py 119 additions, 0 deletionsoneflow/python/nn/modules/sparse.py
- oneflow/python/test/modules/test_gather.py 99 additions, 0 deletionsoneflow/python/test/modules/test_gather.py
- oneflow/python/test/modules/test_sparse.py 69 additions, 0 deletionsoneflow/python/test/modules/test_sparse.py
Please register or sign in to comment