Fix Checkfail in MatrixDiagV3
This PR is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you. Done the suggested changes.
The Op raw_ops.MatrixDiagV3 terminates due to checkfail when the input `diagonal` is of `rank=1` and `k[0] != k[1]`.

This happens at this line below which is trying to access `diagonal_shape.dim_size(-1)` when rank=1 . Currently API will not work with Rank=1 and k[0] != k[1] due to this check fail.

https://github.com/tensorflow/tensorflow/blob/e193d8ea7776ef5c6f5d769b6fb9c070213e737a/tensorflow/core/kernels/linalg/matrix_diag_op.cc#L241

Proposed a fix for same.

Fixes #62984 .