enable_dygraph

paddle.fluid.dygraph.enable_dygraph(place=None)[源代码]

该接口打开动态图模式。

注解

推荐使用 enable_imperative 。

参数

• place (fluid.CPUPlace 或 fluid.CUDAPlace，可选) - 执行动态图的设备数目。若为None，则设备根据paddle的编译方式决定。默认值为 None 。

返回

无

代码示例

import paddle.fluid as fluid
import numpy as np
fluid.dygraph.enable_dygraph()  # Now we are in dygraph mode
print(fluid.in_dygraph_mode())  # True
fluid.dygraph.disable_dygraph()
print(fluid.in_dygraph_mode())  # False