Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 20 additions & 0 deletions docs/api/paddle/get_rng_state_cn.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
.. _cn_api_paddle_get_rng_state:

get_rng_state
-------------------------------
.. py:function:: paddle.get_rng_state(device=None)

获取指定设备的随机数生成器的所有随机状态。

参数
::::::::::::
- **device** (str) - 此参数决定了具体的运行设备。可以是 ``cpu``、``gpu``、``xpu``。默认值为 None。如果为 None,则返回当前设备(由 ``set_device`` 指定)的生成器。

返回
::::::::::::
- GeneratorState:对象。

代码示例
::::::::::::

COPY-FROM: paddle.get_rng_state
23 changes: 23 additions & 0 deletions docs/api/paddle/set_rng_state_cn.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
.. _cn_api_paddle_set_rng_state:

set_rng_state
-------------------------------

.. py:function:: paddle.set_rng_state(state_list, device=None)

为所有设备生成器设置生成器状态。

参数
::::::::::::
- **state_list** (list|tuple) - 要设置回设备生成器的设备状态。state_list 是从 ``get_rng_state()`` 获取的。
- **device** (str) - 此参数决定了具体的运行设备。可以是 ``cpu``、``gpu``、``xpu``。默认值为 None。如果为 None,则返回当前设备(由 ``set_device`` 指定)的生成器。

返回
::::::::::::
- 无。


代码示例
::::::::::::

COPY-FROM: paddle.set_rng_state