frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Docs: update default device description (#101283)

Browse Source 
Closes #101274

Pull Request resolved: https://github.com/pytorch/pytorch/pull/101283
Approved by: https://github.com/albanD
This commit is contained in:
Snorf Yang
2023-05-16 17:07:26 +00:00
committed by PyTorch MergeBot
parent 010763be9a
commit 2a3e45a2a8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/_torch_docs.py
View File

@ -996,7 +996,7 @@ Args:
{dtype}
device (:class:`torch.device`, optional): the device of the constructed tensor. If None and data is a tensor
then the device of data is used. If None and data is not a tensor then
the result tensor is constructed on the CPU.
the result tensor is constructed on the current device.
Example::
@ -9273,7 +9273,7 @@ Keyword args:
{dtype}
device (:class:`torch.device`, optional): the device of the constructed tensor. If None and data is a tensor
then the device of data is used. If None and data is not a tensor then
the result tensor is constructed on the CPU.
the result tensor is constructed on the current device.
{requires_grad}
{pin_memory}