Add a note to torch.round docs to indicate that the return type of out is the same as input.dtype #89056
Labels
actionable
module: docs
Related to our documentation, both in docs/ and docblocks
module: python frontend
For issues relating to PyTorch's Python frontend
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
馃摎 The doc issue
One might expect that the output of torch.round will be an int tensor but that's not the case. Let's add a note to the doc to clarify that this is intended since it can be a little bit surprising for the user.
Suggest a potential alternative/fix
No response
cc @svekars @carljparker @albanD
The text was updated successfully, but these errors were encountered: