export function identityMatrix(n: number): number[][] {
  if(n === 1) return [[1]]
  let matrix = Array.from({length: n}, () => new Array(n).fill(0))
  for(let i=0; i < n; i++){
    matrix[i][i] = 1;
  }
  console.log(matrix)
  return matrix;
}