Dev type as module (#5349)
* refine and add test case
* support ellipsis type slice
* refine
* refine
* support slice assign ellipsis type
* refine
* register fn to localtensor
* add type_as module
* add long module
* add more test
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- docs/source/experimental.rst 2 additions, 0 deletionsdocs/source/experimental.rst
- oneflow/python/nn/modules/tensor_ops.py 94 additions, 0 deletionsoneflow/python/nn/modules/tensor_ops.py
- oneflow/python/test/modules/test_tensor_ops.py 64 additions, 0 deletionsoneflow/python/test/modules/test_tensor_ops.py
Please register or sign in to comment