experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Small documentation clarification.

Browse Source 
PiperOrigin-RevId: 336142016
Change-Id: I0be807ff176bc32af473c49f12549755f79138d4
This commit is contained in:
Brian Patton 2020-10-08 12:33:37 -07:00 committed by TensorFlower Gardener
parent a42d76b6c8
commit a3934c53e2
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/ops/linalg_ops.py
View File

@ -151,6 +151,9 @@ def matrix_triangular_solve(matrix, rhs, lower=True, adjoint=False, name=None):
def cholesky_solve(chol, rhs, name=None):
"""Solves systems of linear eqns `A X = RHS`, given Cholesky factorizations.
Specifically, returns `X` from `A X = RHS`, where `A = L L^T`, `L` is the
`chol` arg and `RHS` is the `rhs` arg.
```python
# Solve 10 separate 2x2 linear systems:
A = ... # shape 10 x 2 x 2