TensorFlow函数:tf.matrix_solve
2018-11-01 17:31 更新
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].