enable_prim¶

paddle.incubate.autograd. enable_prim ( ) [源代码] ¶

注解

只支持在静态图模式下使用。

开启基于自动微分基础算子的自动微分机制。

返回¶

无

代码示例¶

          import paddle
from paddle.incubate.autograd import enable_prim, prim_enabled

paddle.enable_static()
enable_prim()

print(prim_enabled()) # True