tf.matrix_solve
tf.matrix_solve
tf.matrix_solve
matrix_solve( matrix, rhs, adjoint=None, name=None )
Defined in tensorflow/python/ops/gen_linalg_ops.py
.
See the guide: Math > Matrix Math Functions
Solves systems of linear equations.
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
: ATensor
. Must be one of the following types:float64
,float32
,complex64
,complex128
. Shape is[..., M, M]
. -
rhs
: ATensor
. Must have the same type asmatrix
. Shape is[..., M, K]
. -
adjoint
: An optionalbool
. Defaults toFalse
. Boolean indicating whether to solve withmatrix
or its (block-wise) adjoint. -
name
: A name for the operation (optional).
Returns:
A Tensor
. Has the same type as matrix
. Shape is [..., M, K]
.
© 2017 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/api_docs/python/tf/matrix_solve