guard_cn.rst 983 字节
Newer Older
H
Hao Wang 已提交
1 2 3 4 5
.. _cn_api_fluid_dygraph_guard:

guard
-------------------------------

6

H
Hao Wang 已提交
7 8
.. py:function:: paddle.fluid.dygraph.guard(place=None)

S
swtkiwi 已提交
9 10 11 12
:api_attr: 命令式编程模式(动态图)



13
通过with语句创建一个dygraph运行的context,执行context代码。
H
Hao Wang 已提交
14 15

参数:
16
    - **place** (fluid.CPUPlace|fluid.CUDAPlace, 可选) –  动态图执行的设备,可以选择cpu,gpu,如果用户未制定,则根据用户paddle编译的方式来选择运行的设备,如果编译的cpu版本,则在cpu上运行,如果是编译的gpu版本,则在gpu上运行。默认值:None。
H
Hao Wang 已提交
17 18 19 20 21 22 23 24 25 26 27

返回: None

**代码示例**

.. code-block:: python

    import numpy as np
    import paddle.fluid as fluid

    with fluid.dygraph.guard():
28
        inp = np.ones([3, 1024], dtype='float32')
H
Hao Wang 已提交
29
        t = fluid.dygraph.base.to_variable(inp)
30 31 32 33
        linear1 = fluid.Linear(1024, 4, bias_attr=False)
        linear2 = fluid.Linear(4, 4)
        ret = linear1(t)
        dy_ret = linear2(ret)
H
Hao Wang 已提交
34 35