Dev merge modules part2 (#4776)
* add logsoftmax module * add sigmoid module * add maskedfill module * format * add unsquuze module * add eq module * add arange module * fix arange module * add softmax module * refine * update logsoftmax module * update eq module * update eq module * refine * refine according to comments * refine softmax test case * fix unsqueeze module * refine according to comments * refine * format * format Co-authored-by:Yinggang Wang <wyg19970408@gmail.com> Co-authored-by:
Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/nn/modules/activation.py 232 additions, 11 deletionsoneflow/python/nn/modules/activation.py
- oneflow/python/nn/modules/arange.py 70 additions, 0 deletionsoneflow/python/nn/modules/arange.py
- oneflow/python/nn/modules/eq.py 75 additions, 0 deletionsoneflow/python/nn/modules/eq.py
- oneflow/python/nn/modules/masked_fill.py 84 additions, 0 deletionsoneflow/python/nn/modules/masked_fill.py
- oneflow/python/nn/modules/unsqueeze.py 65 additions, 0 deletionsoneflow/python/nn/modules/unsqueeze.py
- oneflow/python/test/modules/test_activation.py 116 additions, 0 deletionsoneflow/python/test/modules/test_activation.py
- oneflow/python/test/modules/test_arange.py 47 additions, 0 deletionsoneflow/python/test/modules/test_arange.py
- oneflow/python/test/modules/test_eq.py 51 additions, 0 deletionsoneflow/python/test/modules/test_eq.py
- oneflow/python/test/modules/test_masked_fill.py 72 additions, 0 deletionsoneflow/python/test/modules/test_masked_fill.py
- oneflow/python/test/modules/test_unsqueeze.py 51 additions, 0 deletionsoneflow/python/test/modules/test_unsqueeze.py
Please register or sign in to comment