add experimental apis (#4817)
* add experimental apis
* merge master fix conflict
* revert flow._oneflow_internal.dtype to flow.dtype
* refine
* fix test optimizer
* update module docs
* fix unit tests
* fix matmul module test
* fix adamw and rmsprop tests
* fix crossentropy loss grad
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/framework/ops.py 2 additions, 1 deletiononeflow/python/framework/ops.py
- oneflow/python/framework/tensor.py 3 additions, 3 deletionsoneflow/python/framework/tensor.py
- oneflow/python/nn/modules/activation.py 21 additions, 11 deletionsoneflow/python/nn/modules/activation.py
- oneflow/python/nn/modules/arange.py 2 additions, 1 deletiononeflow/python/nn/modules/arange.py
- oneflow/python/nn/modules/argmax.py 5 additions, 4 deletionsoneflow/python/nn/modules/argmax.py
- oneflow/python/nn/modules/batchnorm.py 5 additions, 3 deletionsoneflow/python/nn/modules/batchnorm.py
- oneflow/python/nn/modules/cast.py 1 addition, 1 deletiononeflow/python/nn/modules/cast.py
- oneflow/python/nn/modules/concat.py 3 additions, 2 deletionsoneflow/python/nn/modules/concat.py
- oneflow/python/nn/modules/constant.py 13 additions, 9 deletionsoneflow/python/nn/modules/constant.py
- oneflow/python/nn/modules/container.py 6 additions, 1 deletiononeflow/python/nn/modules/container.py
- oneflow/python/nn/modules/conv.py 2 additions, 1 deletiononeflow/python/nn/modules/conv.py
- oneflow/python/nn/modules/dropout.py 2 additions, 1 deletiononeflow/python/nn/modules/dropout.py
- oneflow/python/nn/modules/eq.py 3 additions, 2 deletionsoneflow/python/nn/modules/eq.py
- oneflow/python/nn/modules/exp.py 2 additions, 1 deletiononeflow/python/nn/modules/exp.py
- oneflow/python/nn/modules/expand.py 4 additions, 3 deletionsoneflow/python/nn/modules/expand.py
- oneflow/python/nn/modules/flatten.py 4 additions, 2 deletionsoneflow/python/nn/modules/flatten.py
- oneflow/python/nn/modules/greater.py 3 additions, 2 deletionsoneflow/python/nn/modules/greater.py
- oneflow/python/nn/modules/less.py 3 additions, 2 deletionsoneflow/python/nn/modules/less.py
- oneflow/python/nn/modules/linear.py 3 additions, 1 deletiononeflow/python/nn/modules/linear.py
- oneflow/python/nn/modules/loss.py 20 additions, 18 deletionsoneflow/python/nn/modules/loss.py
Please register or sign in to comment