diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py index 1be99cb4a701ee6a16a550745cf61a3fcc03f31b..e136804f8a12b200e861abb3658a8f5672793c67 100644 --- a/python/paddle/fluid/__init__.py +++ b/python/paddle/fluid/__init__.py @@ -89,6 +89,8 @@ from .io import save, load, load_program_state, set_program_state from .dygraph.checkpoint import save_dygraph, load_dygraph from .dygraph.varbase_patch_methods import monkey_patch_varbase Tensor = LoDTensor +enable_imperative = enable_dygraph +disable_imperative = disable_dygraph __all__ = framework.__all__ + executor.__all__ + \ trainer_desc.__all__ + transpiler.__all__ + \ @@ -104,6 +106,8 @@ __all__ = framework.__all__ + executor.__all__ + \ 'dygraph', 'enable_dygraph', 'disable_dygraph', + 'enable_imperative', + 'disable_imperative', 'transpiler', 'nets', 'optimizer',