Return the n×n identity matrix: 1s on the main diagonal, 0s elsewhere.
Input
3
Output
tensor([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])
Sign in to write and run your own code.
solve(...)