torch.CholeskyShape

export type CholeskyShape<S extends Shape> =
  ValidateBatchedSquareMatrix<S> extends { readonly __isShapeError: true }
    ? ValidateBatchedSquareMatrix<S> // Propagate error
    : S;

Sextends Shape

Output shape of Cholesky decomposition. Input must be square matrix (or batched square matrices).