add kldivloss module (#5155)
* add kldivloss module
* support for larger range of target values
* add doctest
* separate testcase
* delete deprecated argument
* fix zeros device bug
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- docs/source/experimental.rst 1 addition, 0 deletionsdocs/source/experimental.rst
- oneflow/python/nn/modules/loss.py 137 additions, 34 deletionsoneflow/python/nn/modules/loss.py
- oneflow/python/ops/nn_ops.py 1 addition, 0 deletionsoneflow/python/ops/nn_ops.py
- oneflow/python/test/modules/test_kldivloss.py 118 additions, 0 deletionsoneflow/python/test/modules/test_kldivloss.py
Please register or sign in to comment