forked from mindspore-Ecosystem/mindspore
27 lines
1.3 KiB
ReStructuredText
27 lines
1.3 KiB
ReStructuredText
|
mindspore.ops.MatrixSolve
|
|||
|
|
=========================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.ops.MatrixSolve(adjoint=False)
|
|||
|
|
|
|||
|
|
求解线性方程组。
|
|||
|
|
|
|||
|
|
参数：
|
|||
|
|
- **adjoint** (bool，可选) - 指明是用矩阵求解还是用其（逐块）伴随求解。默认值：False。
|
|||
|
|
|
|||
|
|
输入：
|
|||
|
|
- **matrix** (Tensor) - Tensor，线性方程组系数组成的矩阵，其shape为 :math:`[..., M, M]` 。
|
|||
|
|
- **rhs** (Tensor) - Tensor，线性方程组结果值组成的矩阵，其shape为 :math:`[..., M, K]` ， `rhs` 与 `matrix` 的类型必须一致。
|
|||
|
|
|
|||
|
|
输出：
|
|||
|
|
Tensor，线性方程组解组成的矩阵，与 `rhs` 的shape及类型均相同。
|
|||
|
|
|
|||
|
|
异常：
|
|||
|
|
- **TypeError** - 如果 `adjoint` 不是bool型。
|
|||
|
|
- **TypeError** - 如果 `matrix` 的类型不是以下之一：
|
|||
|
|
mstype.float16、mstype.float32、mstype.float64、mstype.complex64、mstype.complex128。
|
|||
|
|
- **TypeError** - 如果 `rhs` 与 `matrix` 的类型不一致。
|
|||
|
|
- **ValueError** - 如果 `matrix` 的秩小于2。
|
|||
|
|
- **ValueError** - 如果 `matrix` 和 `rhs` 的维度不同。
|
|||
|
|
- **ValueError** - 如果 `matrix` 的最内两维不同。
|
|||
|
|
- **ValueError** - 如果 `rhs` 的最内两维与 `matrix` 不匹配。
|