Add 'nn.Mish' module and corresponding functions (#5310)
* Add 'nn.Mish' module and corresponding functions * Improve the docs and unit test * auto format by CI * Fix the indent problems Co-authored-by:Yao Chi <later@usopp.net> Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org>
Showing
- docs/source/experimental.rst 3 additions, 0 deletionsdocs/source/experimental.rst
- oneflow/python/nn/modules/activation.py 69 additions, 0 deletionsoneflow/python/nn/modules/activation.py
- oneflow/python/test/modules/test_activation.py 55 additions, 5 deletionsoneflow/python/test/modules/test_activation.py
- oneflow/python/test/tensor/test_tensor.py 24 additions, 0 deletionsoneflow/python/test/tensor/test_tensor.py
Please register or sign in to comment