diff --git a/doc/prune_api.md b/doc/prune_api.md
index a9a0ea7c91f37fa43da2707326580ff618f0d9d5..99522dad849b2ffde8ab39fbe7af15013bb0cf22 100644
--- a/doc/prune_api.md
+++ b/doc/prune_api.md
@@ -63,9 +63,11 @@ for block in program.blocks:
 
 **示例：**
 
+点击[AIStudio](https://aistudio.baidu.com/aistudio/projectDetail/200786)执行以下示例代码。
 ```
 
 import paddle.fluid as fluid
+from paddle.fluid.param_attr import ParamAttr
 from paddleslim.prune import Pruner
 
 def conv_bn_layer(input,