Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion docs/api/paddle/autograd/saved_tensors_hooks_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ saved_tensors_hooks 用于动态图,注册一对 pack / unpack hook,用于
**参数**

- **pack_hook** (function) – 当某个算子的前向执行时,存在 Tensor 需要保留给反向计算梯度使用时, ``pack_hook`` 将会被调用。 ``pack_hook`` 可以将 Tensor 临时存放到内存或者硬盘上。 ``pack_hook`` 的输入是 1 个要被保留的 Tensor。 ``pack_hook`` 的输出是恢复被保留 Tensor 所需要的信息。当 ``PyLayerContext.save_for_backward`` 被调用时, ``pack_hook`` 也会被调用。如果一个 Tensor 是 no need buffer 的(即反向不需要数据内容,只需要数据的 meta 信息), ``pack_hook`` 则不会被调用。只有需要保留的 Tensor 带有 LoD 信息, ``pack_hook`` 才会被调用。
- **unpack_hook** (function) – 当反向执行,需要用到前向保留的 Tensor 时, ``unpack_hook`` 会被调用 ``unpack_hook`` 的输入是 ``pack_hook `` 输出的用于恢复 Tensor 所需的信息。 ``unpack_hook`` 的输出是恢复后的 Tensor,这个 Tensor 的数据内容应该和 ``pack_hook`` 的输入严格一致。
- **unpack_hook** (function) – 当反向执行,需要用到前向保留的 Tensor 时, ``unpack_hook`` 会被调用 ``unpack_hook`` 的输入是 ``pack_hook`` 输出的用于恢复 Tensor 所需的信息。 ``unpack_hook`` 的输出是恢复后的 Tensor,这个 Tensor 的数据内容应该和 ``pack_hook`` 的输入严格一致。

**返回**

Expand Down