mindspore.mint.triangular_solve =============================== .. py:function:: mindspore.mint.triangular_solve(b, A, upper=True, transpose=False, unitriangular=False) 求解正上三角形或下三角形可逆矩阵 `A` 和包含多个元素的右侧边 `b` 的方程组的解。 用符号表示,它求解方程 :math:`A X = b`,并假设矩阵 `A` 是一个方阵,且为上三角矩阵(如果 ``upper = False``,则为下三角矩阵),并且其对角线上没有零元素。 .. warning:: 这是一个实验性API,后续可能修改或删除。 参数: - **b** (Tensor) - shape是 :math:`(*, M, K)` 的tensor,其中*表示任意数量的维度。 - **A** (Tensor) - shape是 :math:`(*, M, M)` 的tensor,其中*表示任意数量的维度。 - **upper** (bool,可选) - 矩阵 `A` 是否为上三角矩阵。如果为 ``False``,则矩阵 `A` 为下三角矩阵。默认 ``True``。 - **transpose** (bool,可选) - 求解方程 :math:`op(A) X = b`。其中如果此标志为 ``True``,则 :math:`op(A) = A^T`;如果为 ``False``,则 :math:`op(A) = A`。默认 ``False``。 - **unitriangular** (bool,可选) - 矩阵 `A` 是否为单位三角矩阵。如果为 ``True``,则假设矩阵 `A` 的对角线元素为 1,并且不会从矩阵 `A` 中引用这些对角线元素。默认 ``False``。 返回: 包含 `X` 和 `A` 的tuple。 异常: - **ValueError** - 如果 `b` 或者 `A` 的维度不在 :math:`[2, 6]` 范围内。 - **ValueError** - 如果 `b` 和 `A` 的shape不匹配。