Skip to content

Commit 75b7b7b

Browse files
fix rendering of api doc saved_tensors_hooks_cn after reverting #7462 (#7504)
1 parent 19aa568 commit 75b7b7b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/api/paddle/autograd/saved_tensors_hooks_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ saved_tensors_hooks 用于动态图,注册一对 pack / unpack hook,用于
1212
**参数**
1313

1414
- **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`` 才会被调用。
15-
- **unpack_hook** (function) – 当反向执行,需要用到前向保留的 Tensor 时, ``unpack_hook`` 会被调用 ``unpack_hook`` 的输入是 ``pack_hook `` 输出的用于恢复 Tensor 所需的信息。 ``unpack_hook`` 的输出是恢复后的 Tensor,这个 Tensor 的数据内容应该和 ``pack_hook`` 的输入严格一致。
15+
- **unpack_hook** (function) – 当反向执行,需要用到前向保留的 Tensor 时, ``unpack_hook`` 会被调用 ``unpack_hook`` 的输入是 ``pack_hook`` 输出的用于恢复 Tensor 所需的信息。 ``unpack_hook`` 的输出是恢复后的 Tensor,这个 Tensor 的数据内容应该和 ``pack_hook`` 的输入严格一致。
1616

1717
**返回**
1818

0 commit comments

Comments
 (0)