tf.keras.ops.solve

Solves a linear system of equations given by a x = b.

View aliases

Main aliases

tf.keras.ops.linalg.solve

tf.keras.ops.solve(
    a, b
)

Args
a	A tensor of shape (..., M, M) representing the coefficients matrix.
b	A tensor of shape (..., M) or (..., M, N) represeting the right-hand side or "dependent variable" matrix.

Returns
A tensor of shape (..., M) or (..., M, N) representing the solution of the linear system. Returned shape is identical to b.