From 8d33c91e0019215f1eb169fc2873ae89b01c8a62 Mon Sep 17 00:00:00 2001 From: wangshuide2020 <7511764+wangshuide2020@user.noreply.gitee.com> Date: Sat, 27 Feb 2021 17:42:30 +0800 Subject: [PATCH] fix the format error of MatrixSetDiag operator. --- mindspore/nn/layer/basic.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/nn/layer/basic.py b/mindspore/nn/layer/basic.py index f7115e1848..0b2b1fe612 100644 --- a/mindspore/nn/layer/basic.py +++ b/mindspore/nn/layer/basic.py @@ -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: