tf.linalg.solve  |  TensorFlow v2.16.1 (original) (raw)

tf.linalg.solve

Stay organized with collections Save and categorize content based on your preferences.

Solves systems of linear equations.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.linalg.solve, tf.compat.v1.matrix_solve

tf.linalg.solve(
    matrix: Annotated[Any, TV_MatrixSolve_T],
    rhs: Annotated[Any, TV_MatrixSolve_T],
    adjoint: bool = False,
    name=None
) -> Annotated[Any, TV_MatrixSolve_T]

Used in the notebooks

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 satisfiesadjoint(matrix[..., :, :]) * output[..., :, :] = rhs[..., :, :].