Add int
type to device
parameter of torch.set_default_device() on the doc
#126646
Labels
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
The doc of
torch.set_default_device()
explainsdevice
parameter but it doesn't haveint
type as shown below:Actually,
device
parameter withint
type works as shown below:Suggest a potential alternative/fix
The doc should also say
int
type as shown below:cc @svekars @brycebortree @albanD
The text was updated successfully, but these errors were encountered: