阅读(7.7k) 书签 (0)

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].