未验证 提交 9283a400 编写于 作者: Y yaoxuefeng 提交者: GitHub

add cuda genrator get/set_cuda_rng_state doc (#2542)

上级 daeee51f
...@@ -24,6 +24,8 @@ paddle.framework ...@@ -24,6 +24,8 @@ paddle.framework
framework_cn/global_scope_cn.rst framework_cn/global_scope_cn.rst
framework_cn/gradients_cn.rst framework_cn/gradients_cn.rst
framework_cn/manual_seed_cn.rst framework_cn/manual_seed_cn.rst
framework_cn/get_cuda_rng_state_cn.rst
framework_cn/set_cuda_rng_state_cn.rst
framework_cn/name_scope_cn.rst framework_cn/name_scope_cn.rst
framework_cn/ParallelExecutor_cn.rst framework_cn/ParallelExecutor_cn.rst
framework_cn/ParamAttr_cn.rst framework_cn/ParamAttr_cn.rst
......
.. _cn_api_paddle_framework_get_cuda_rng_state:
get_cuda_rng_state
-------------------------------
.. py:function:: paddle.framework.get_cuda_rng_state()
获取cuda随机数生成器的状态信息
参数:
返回:
GeneratorState:对象
**代码示例**:
.. code-block:: python
import paddle
sts = paddle.get_cuda_rng_state()
.. _cn_api_paddle_framework_set_cuda_rng_state:
set_cuda_rng_state
-------------------------------
.. py:function:: paddle.framework.set_cuda_rng_state(state_list)
设置cuda随机数生成器的状态信息
参数:
- **state_list** (list [GeneratorState]) - 需要设置的随机数生成器状态信息列表,通过get_cuda_rng_state()获取。
返回:
**代码示例**:
.. code-block:: python
import paddle
sts = paddle.get_cuda_rng_state()
paddle.set_cuda_rng_state(sts)
...@@ -96,6 +96,8 @@ paddle ...@@ -96,6 +96,8 @@ paddle
paddle_cn/logsumexp_cn.rst paddle_cn/logsumexp_cn.rst
paddle_cn/log_cn.rst paddle_cn/log_cn.rst
paddle_cn/manual_seed_cn.rst paddle_cn/manual_seed_cn.rst
paddle_cn/get_cuda_rng_state_cn.rst
paddle_cn/set_cuda_rng_state_cn.rst
paddle_cn/masked_select_cn.rst paddle_cn/masked_select_cn.rst
paddle_cn/matmul_cn.rst paddle_cn/matmul_cn.rst
paddle_cn/max_cn.rst paddle_cn/max_cn.rst
......
.. _cn_api_paddle_cn_get_cuda_rng_state:
get_cuda_rng_state
-------------------------------
:doc_source: paddle.framework.get_cuda_rng_state
.. _cn_api_paddle_cn_set_cuda_rng_state:
set_cuda_rng_state
-------------------------------
:doc_source: paddle.framework.set_cuda_rng_state
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册