Missed TensorFlow World? Check out the recap. Learn more

tf.linalg.solve

TensorFlow 1 version

Defined in generated file: python/ops/gen_linalg_ops.py

Solves systems of linear equations.

Aliases:

tf.linalg.solve(
    matrix,
    rhs,
    adjoint=False,
    name=None
)

Matrix is a tensor of shape [..., M, M] whose inner-most 2 dimensions form square matrices. Rhs is a tensor of shape [..., M, K]. The output is a tensor shape [..., M, K]. If adjoint is False then each output matrix satisfies matrix[..., :, :] * output[..., :, :] = rhs[..., :, :]. If adjoint is True then each output matrix satisfies adjoint(matrix[..., :, :]) * output[..., :, :] = rhs[..., :, :].

Args:

  • matrix: A Tensor. Must be one of the following types: float64, float32, half, complex64, complex128. Shape is [..., M, M].
  • rhs: A Tensor. Must have the same type as matrix. Shape is [..., M, K].
  • adjoint: An optional bool. Defaults to False. Boolean indicating whether to solve with matrix or its (block-wise) adjoint.
  • name: A name for the operation (optional).

Returns:

A Tensor. Has the same type as matrix.