DOC: Docstring of np.diagonal to reflect edits of PR #5358

These should have been included with PR #5358.