"Either torch.float32
or torch.float64
" should be removed from the doc of d
parameter of torch.set_default_dtype()
#126647
Labels
actionable
easy
label for easy docathon tasks
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 explains
d
parameter but it only says Eithertorch.float32
ortorch.float64
as shown below:Actually,
d
parameter withtorch.float16
,torch.half
,torch.bfloat16
,torch.float
ortorch.double
type also works as shown below:Suggest a potential alternative/fix
So,
Either torch.float32 or torch.float64
should be removed from the doc should sayint
type as shown below:cc @svekars @brycebortree @albanD
The text was updated successfully, but these errors were encountered: