paddle.autograd

paddle.autograd 目录下包含飞桨框架支持的自动微分相关的 API 和类。具体如下:

自动微分相关 API

API 名称

API 功能

计算给定的 Tensors 的反向梯度

计算因变量 ys 对 自变量 xs 的海森矩阵

计算因变量 ys 对 自变量 xs 的雅可比矩阵

用于动态图中为保存的 Tensor 注册一对 pack / unpack hook

自动微分相关辅助类

类名称

辅助类功能

通过创建 PyLayer 子类的方式实现 Python 端自定义算子

PyLayerContext 对象能够辅助 PyLayer 实现某些功能