torch.ValidateSquareMatrix

export type ValidateSquareMatrix<S extends Shape> = S extends readonly [
  infer R extends number,
  infer C extends number,
]
  ? R extends C
    ? R // Valid square matrix, return the dimension
    : linalg_error_not_square_matrix<R, C>
  : number extends S['length']
    ? number // Dynamic shape
    : linalg_error_requires_2d<S['length']>;

Sextends Shape

Validate shape is a 2D square matrix. Returns the square size if valid, error otherwise.