update warning for irreversible input

This commit is contained in:
zhujingxuan 2022-07-09 17:19:12 +08:00
parent 93573bd367
commit 1c2f9b0475
2 changed files with 7 additions and 0 deletions

View File

@ -11,6 +11,9 @@ mindspore.ops.matrix_solve
&adjoint(matrix[..., M, M]) * x[..., M, K] = rhs[..., M, K]
\end{aligned}
.. warning::
- 当平台为GPU时如果 `matrix` 中的矩阵不可逆,将产生错误或者返回一个未知结果。
**参数:**
- **matrix** (Tensor) - 输入Tensor shape 为 :math:`[..., M, M]`
@ -30,3 +33,4 @@ mindspore.ops.matrix_solve
- **ValueError** - `rhs` 的维度与 `matrix` 的维度不相等。
- **ValueError** - `matrix` 最内侧的两个维度不相等。
- **ValueError** - `rhs` 最内侧的两个维度和 `matrix` 不能匹配。
- **ValueError** - `matrix` 中的矩阵不可逆。

View File

@ -2185,6 +2185,8 @@ def matrix_solve(matrix, rhs, adjoint=False):
&adjoint(matrix[..., M, M]) * x[..., M, K] = rhs[..., M, K]
\end{aligned}
.. warning::
On GPU, if the matrix is irreversible, an error may be reported or an unknown result may be returned.
Args:
matrix (Tensor): The shape of tensor is :math:`[..., M, M]`.
@ -2203,6 +2205,7 @@ def matrix_solve(matrix, rhs, adjoint=False):
ValueError: If the dimension of `matrix` is not the same as `rhs`.
ValueError: If the inner-most 2 dimension of `matrix` is not the same.
ValueError: If the inner-most 2 dimension of `rhs` does not match `matrix`.
ValueError: If the `matrix` is irreversible.
Supported Platforms:
``GPU`` ``CPU``