eye

odl.eye(impl, n_rows, n_cols=None, k=0, dtype=None, device=None)

Returns a two-dimensional array with ones on the kth diagonal and zeros elsewhere.