Fix slice bug (#5117)
* align squeeze module to torch * add squeeze for slice * refine as comments * refine as comments * refine * refine * redfine * refine test case * refine * refine conv * fix conv slice * refine 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/framework/tensor.py 10 additions, 1 deletiononeflow/python/framework/tensor.py
- oneflow/python/nn/modules/conv.py 4 additions, 5 deletionsoneflow/python/nn/modules/conv.py
- oneflow/python/nn/modules/slice.py 1 addition, 1 deletiononeflow/python/nn/modules/slice.py
- oneflow/python/nn/modules/squeeze.py 8 additions, 1 deletiononeflow/python/nn/modules/squeeze.py
- oneflow/python/test/modules/test_slice.py 41 additions, 1 deletiononeflow/python/test/modules/test_slice.py
- oneflow/python/test/modules/test_squeeze.py 8 additions, 0 deletionsoneflow/python/test/modules/test_squeeze.py
Please register or sign in to comment