init of op diag (#5298)
* init of op diag * init of op diag * modify op diag * merge testcase * delete no use tensor diag * amend diag api docs * resolve conficts * resolve confict * resolve confict * resolve confict * auto format by CI * add tensor.diag * auto format by CI * amend API description * add non_square test * amend diag API description * amend diag API description * compact test code * compact code * auto format by CI * amend API description * amend API description * motify functional API * motify test_diag * amend diag functional API * amend diag docstring * auto format by CI * amend code standards * auto format by CI Co-authored-by:oneflow-ci-bot <ci-bot@oneflow.org> 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/core/autograd/gradient_funcs/diag.cpp 72 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/diag.cpp
- oneflow/core/functional/functional_api.yaml 9 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/array_functor.cpp 31 additions, 0 deletionsoneflow/core/functional/impl/array_functor.cpp
- oneflow/python/nn/modules/diag.py 86 additions, 0 deletionsoneflow/python/nn/modules/diag.py
- oneflow/python/ops/diag_ops.py 4 additions, 2 deletionsoneflow/python/ops/diag_ops.py
- oneflow/python/test/modules/test_diag.py 162 additions, 0 deletionsoneflow/python/test/modules/test_diag.py
Please register or sign in to comment