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

Update description for torch.random.fork_rng (#151881)

Browse Source 
As the title stated.

Related ISSUE:
https://github.com/pytorch/pytorch/issues/151784
Pull Request resolved: https://github.com/pytorch/pytorch/pull/151881
Approved by: https://github.com/albanD
This commit is contained in:
FFFrog
2025-04-23 14:23:57 +08:00
committed by PyTorch MergeBot
parent 05114679b7
commit a560216abb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/random.py
View File

@ -143,8 +143,8 @@ def fork_rng(
enabled (bool): if ``False``, the RNG is not forked. This is a convenience
argument for easily disabling the context manager without having
to delete it and unindent your Python code under it.
device_type (str): device type str, default is `cuda`. As for custom device,
see details in [Note: support the custom device with privateuse1]
device_type (str): device type str, default is `cuda`. As for supported device,
see details in :ref:`accelerator<accelerators>`
"""
if device_type == "meta":