


A = A @ A.mT + torch.eye(2)  # batch of symmetric positive-definite matrices
      >>> L = torch.linalg.cholesky(A)
      >>> torch.dist(L @ L.mT, A)
      tensor(5.8747e-16, dtype=torch.float64)


