Computes the Cholesky decomposition of one or more square matrices.
Compat aliases for migration
See Migration guide for more details.
tf.raw_ops.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
[..., :, :].
Note: The gradient computation on GPU is faster for large matrices but not for large batch dimensions when the submatrices are small. In this case it might be faster to use the CPU.
| || A |
| ||A name for the operation (optional).|
| A |
© 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.