tf.matrix_solve 函数

matrix_solve(
    matrix,
    rhs,
    adjoint=None,
    name=None
)

参考指南：数学函数>矩阵数学函数

求解线性方程组.

Matrix 是一个形状为 [..., M, M] 的张量,并且该张量最内部的 2 维构成方形矩阵.Rhs 是一个形状为 [..., M, K] 的张量.该函数的输出是一个形状为 [..., M, K] 的张量.如果 adjoint 为 False,则每个输出矩阵满足矩阵 [..., :, :] * output[..., :, :] = rhs[..., :, :].如果 adjoint为 True,则每个输出矩阵满足 adjoint(matrix[..., :, :]) * output[..., :, :] = rhs[..., :, :].

参数：

• matrix：张量,必须是下列类型之一：float64,float32,complex64,complex128；并且形状为 [..., M, M].
• rhs：张量,必须与 matrix 具有相同的类型；形状为 [..., M, K].
• adjoint：可选的 bool 值,默认取值为 False.布尔值表示是否用矩阵或其 (block-wise) adjoint 来求解.
• name：操作的名称(可选).

返回值：

该操作返回张量,与 matrix 具有相同的类型；形状是 [..., M, K].