Browse Source

!12704 fix the format error of MatrixSetDiag operator.

From: @wangshuide2020
Reviewed-by: @liangchenghui,@ljl0711
Signed-off-by: @liangchenghui
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
d1b1ad8ad0
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