TensorFlow 2 version |
Computes the Cholesky decomposition of one or more square matrices.
tf.linalg.cholesky(
input, name=None
)
The input is a tensor of shape [..., M, M]
whose inner-most 2 dimensions
form square matrices.
The input has to be symmetric and positive definite. Only the lower-triangular part of the input will be used for this operation. The upper-triangular part will not be read.
The output is a tensor of the same shape as the input
containing the Cholesky decompositions for all input submatrices [..., :, :]
.
Args | |
---|---|
input
|
A Tensor . Must be one of the following types: float64 , float32 , half , complex64 , complex128 .
Shape is [..., M, M] .
|
name
|
A name for the operation (optional). |
Returns | |
---|---|
A Tensor . Has the same type as input .
|