torch.cholesky_solve
torch.cholesky_solve(input, input2, upper=False, *, out=None) -> Tensor
Solves a linear system of equations with a positive semidefinite
   matrix to be inverted given its Cholesky factor matrix u.
If "upper" is "False", u is and lower triangular and c is
   returned such that:
  c = (u u^T)^{{-1}} b

If "upper" is "True" or not provided, u is upper triangular and c
   is returned such that:
  c = (u^T u)^{{-1}} b

torch.cholesky_solve(b, u) can take in 2D inputs b, u or inputs
   that are batches of 2D matrices. If the inputs are batches, then
   returns batched outputs c
Supports real-valued and complex-valued inputs. For the complex-
   valued inputs the transpose operator above is the conjugate
   transpose.
Parameters:
      * input (Tensor) -- input matrix b of size (*, m, k),
        where * is zero or more batch dimensions
  * **input2** (*Tensor*) -- input matrix u of size (*, m, m),
