TensorFlow函數(shù)：tf.matrix_solve

tf.matrix_solve 函數(shù)

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

參考指南：數(shù)學(xué)函數(shù)>矩陣數(shù)學(xué)函數(shù)

求解線性方程組.

Matrix 是一個形狀為 [..., M, M] 的張量,并且該張量最內(nèi)部的 2 維構(gòu)成方形矩陣.Rhs 是一個形狀為 [..., M, K] 的張量.該函數(shù)的輸出是一個形狀為 [..., M, K] 的張量.如果 adjoint 為 False,則每個輸出矩陣滿足矩陣 [..., :, :] * output[..., :, :] = rhs[..., :, :].如果 adjoint為 True,則每個輸出矩陣滿足 adjoint(matrix[..., :, :]) * output[..., :, :] = rhs[..., :, :].

參數(shù)：

• matrix：張量,必須是下列類型之一：float64,float32,complex64,complex128；并且形狀為 [..., M, M].
• rhs：張量,必須與 matrix 具有相同的類型；形狀為 [..., M, K].
• adjoint：可選的 bool 值,默認(rèn)取值為 False.布爾值表示是否用矩陣或其 (block-wise) adjoint 來求解.
• name：操作的名稱(可選).

返回值：

該操作返回張量,與 matrix 具有相同的類型；形狀是 [..., M, K].