paddle.autograd
paddle.autograd 目录下包含飞桨框架支持的自动微分相关的 API 和类。具体如下:
自动微分相关 API
API 名称 |
API 功能 |
|
计算给定的 Tensors 的反向梯度 |
|
计算因变量 ys 对 自变量 xs 的海森矩阵 |
|
计算因变量 ys 对 自变量 xs 的雅可比矩阵 |
|
用于动态图中为保存的 Tensor 注册一对 pack / unpack hook |
自动微分相关辅助类
类名称 |
辅助类功能 |
|
通过创建 PyLayer 子类的方式实现 Python 端自定义算子 |
|
PyLayerContext 对象能够辅助 PyLayer 实现某些功能
|