First input must be 2D. K-th diagonal input must be an integer scalar.