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