import numpy as np
elements = input().split()
int_elements = list(map(int, elements))
result = np.eye(int_elements[0], int_elements[1])
Adjusting the print format to match the expected output
np.set_printoptions(legacy='1.13') # Enables old formatting style
print(result)