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