export function identityMatrix(n: number): number[][] {
  const matrix = Array.from({ length: n }, () => Array(n).fill(0))

  for (let i = 0; i < n; i++) {
    matrix[i][i] = 1
  }

  return matrix;
}