File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -12,7 +12,7 @@ saved_tensors_hooks 用于动态图,注册一对 pack / unpack hook,用于
12
12
**参数 **
13
13
14
14
- **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 `` 的输入严格一致。
16
16
17
17
**返回 **
18
18
You can’t perform that action at this time.
0 commit comments