enable_dygraph

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

该接口打开动态图模式。

参数

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

返回

代码示例

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