[Update] saved_tensors_hooks example (#6700)

megemini · web-flow · commit 1cd89a66e709 · 2024-06-28T10:59:47.000+08:00
diff --git a/docs/api/paddle/autograd/saved_tensors_hooks_cn.rst b/docs/api/paddle/autograd/saved_tensors_hooks_cn.rst
@@ -18,7 +18,12 @@ saved_tensors_hooks 用于动态图，注册一对 pack / unpack hook，用于
 
 无
 
-代码示例
+代码示例 1
 ::::::::::::
 
-COPY-FROM: paddle.autograd.saved_tensors_hooks
+COPY-FROM: paddle.autograd.saved_tensors_hooks:code-example1
+
+代码示例 2
+::::::::::::
+
+COPY-FROM: paddle.autograd.saved_tensors_hooks:code-example2
