Modules chunk (#5324)
* chunk forward * chunk forward * forward nn.chunk * chunk forward * chunk forward * complete module flow.chunk * auto format by CI * flow.chunk/tensor.chunk * resolve confict * auto format by CI * functional api * auto format by CI * doctest.testmod(raise_on_error=True) Co-authored-by:oneflow-ci-bot <ci-bot@oneflow.org> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- docs/source/experimental.rst 2 additions, 0 deletionsdocs/source/experimental.rst
- oneflow/python/nn/modules/chunk.py 139 additions, 0 deletionsoneflow/python/nn/modules/chunk.py
- oneflow/python/nn/modules/slice.py 5 additions, 0 deletionsoneflow/python/nn/modules/slice.py
- oneflow/python/test/modules/test_chunk.py 208 additions, 0 deletionsoneflow/python/test/modules/test_chunk.py
Please register or sign in to comment