Browse Source

fix the format error of MatrixSetDiag operator.

tags/v1.2.0-rc1
wangshuide2020 4 years ago
parent
commit
8d33c91e00
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      mindspore/nn/layer/basic.py

+ 4
- 2
mindspore/nn/layer/basic.py View File

@@ -1008,9 +1008,11 @@ class MatrixSetDiag(Cell):
dimensions :math:`[I, J, K, ..., min(M, N)]`. Then the output is a tensor of rank :math:`k+1` with dimensions
:math:`[I, J, K, ..., M, N]` where:

:math:`output[i, j, k, ..., m, n] = diagnoal[i, j, k, ..., n]\ for\ m == n`
.. math::
output[i, j, k, ..., m, n] = diagnoal[i, j, k, ..., n]\ for\ m == n

:math:`output[i, j, k, ..., m, n] = x[i, j, k, ..., m, n]\ for\ m != n`
.. math::
output[i, j, k, ..., m, n] = x[i, j, k, ..., m, n]\ for\ m != n

Inputs:
- **x** (Tensor) - The batched tensor. Rank k+1, where k >= 1. It can be one of the following data types:


Loading…
Cancel
Save