Solves systems of linear equations.
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 . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/linalg/solve