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