guard

注意:该API仅支持【动态图】模式

通过with语句创建一个dygraph运行的context,执行context代码。

  • 参数:
    • place (fluid.CPUPlace|fluid.CUDAPlace, 可选) – 动态图执行的设备,可以选择cpu,gpu,如果用户未制定,则根据用户paddle编译的方式来选择运行的设备,如果编译的cpu版本,则在cpu上运行,如果是编译的gpu版本,则在gpu上运行。默认值:None。

返回: None

代码示例

  1. import numpy as np
  2. import paddle.fluid as fluid
  3.  
  4. with fluid.dygraph.guard():
  5. inp = np.ones([3, 1024], dtype='float32')
  6. t = fluid.dygraph.base.to_variable(inp)
  7. linear1 = fluid.Linear(1024, 4, bias_attr=False)
  8. linear2 = fluid.Linear(4, 4)
  9. ret = linear1(t)
  10. dy_ret = linear2(ret)