Documentation for `tf.linalg.set_diag()` is missing return information
Actually, I'm unsure whether these docs are for the correct API. I don't see a call to it anywhere in the examples. Ah, they haven't been updated for TF2. They still refer to the old [`matrix_set_diag`](https://www.tensorflow.org/api_docs/python/tf/linalg/set_diag) API. @khatchad  You're right, the official TF documentation for tf.linalg.set_diag() doesn't explicitly mention the return value in the descriptiontf.linalg.set_diag. Please refer the master [branch](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/array_ops.py) for the detailed information.
Thank you! This issue is stale because it has been open for 7 days with no activity. It will be closed if no further activity occurs. Thank you. This issue was closed because it has been inactive for 7 days since being marked as stale. Please reopen if you'd like to work on this further. Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/67255">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/67255">No</a>
 @khatchad Could you please let us know if you have referred to the above [comment](https://github.com/tensorflow/tensorflow/issues/67255#issuecomment-2104241615) and you could see the return information ?
Thank you! > @khatchad Could you please let us know if you have referred to the above [comment](https://github.com/tensorflow/tensorflow/issues/67255#issuecomment-2104241615) and you could see the return information ? Thank you!

Not sure if I understand the question. It's missing here as well:

https://github.com/tensorflow/tensorflow/blob/3825f797eee30efb1a9c33c72f6608709a382928/tensorflow/python/ops/array_ops.py#L2515-L2530 @khatchad,
Could you please try to contribute with the PR for the respective changes to be made in the mentioned document? Thank you!
### Issue type

Documentation Bug

### Have you reproduced the bug with TensorFlow Nightly?

No

### Source

source

### TensorFlow version

2.16.1

### Custom code

No

### OS platform and distribution

_No response_

### Mobile device

_No response_

### Python version

_No response_

### Bazel version

_No response_

### GCC/compiler version

_No response_

### CUDA/cuDNN version

_No response_

### GPU model and memory

_No response_

### Current behavior?

Return information at https://www.tensorflow.org/api_docs/python/tf/linalg/set_diag. I only see args.

### Standalone code to reproduce the issue

It's a documentation "bug."

### Relevant log output

_No response_