From 5eb03ba7efac0fe807c742f7c024f42f928e9523 Mon Sep 17 00:00:00 2001
From: mapingshuo <mps2012@yeah.net>
Date: Thu, 26 Sep 2019 20:26:10 +0800
Subject: [PATCH 1/2] add prune_with_input example

---
 api_examples/paddle/fluid/prune_with_input.py | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 api_examples/paddle/fluid/prune_with_input.py

diff --git a/api_examples/paddle/fluid/prune_with_input.py b/api_examples/paddle/fluid/prune_with_input.py
new file mode 100644
index 0000000..a760fa8
--- /dev/null
+++ b/api_examples/paddle/fluid/prune_with_input.py
@@ -0,0 +1,32 @@
+import paddle.fluid as fluid
+import paddle.fluid.optimizer as optimizer
+import numpy as np
+
+def sample_data():
+   res = []
+   for i in range(2):
+       data = np.random.normal(size=(2,))
+       label = np.random.randint(2, size=(1,))
+       res.append((data, label))
+   return res
+
+x = fluid.layers.data(name='x', shape=[2], dtype='float32')
+label = fluid.layers.data(name="label", shape=[1], dtype="int64")
+
+# define net here
+y = fluid.layers.fc(input=[x], size=2, act="softmax")
+loss = fluid.layers.cross_entropy(input=y, label=label)
+loss = fluid.layers.mean(x=loss)
+
+sgd = fluid.optimizer.SGD(learning_rate=0.01)
+sgd.minimize(loss)
+
+with open("original_program", "w") as f:
+    f.write(str(fluid.default_main_program()))
+
+pruned_program = fluid.default_main_program()._prune_with_input(
+                        feeded_var_names=[y.name, label.name],
+                        targets = [loss])
+
+with open("pruned_program", "w") as f:
+    f.write(str(pruned_program))

From f26d0d1d55e5552100b787a5ae830ef126b12d6b Mon Sep 17 00:00:00 2001
From: mapingshuo <mps2012@yeah.net>
Date: Thu, 26 Sep 2019 22:13:22 +0800
Subject: [PATCH 2/2] add comments

---
 api_examples/paddle/fluid/prune_with_input.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/api_examples/paddle/fluid/prune_with_input.py b/api_examples/paddle/fluid/prune_with_input.py
index a760fa8..edc1157 100644
--- a/api_examples/paddle/fluid/prune_with_input.py
+++ b/api_examples/paddle/fluid/prune_with_input.py
@@ -1,3 +1,22 @@
+# Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# api: paddle.fluid.framework.Program._prune_with_input()
+# env: local
+# device: cpu
+# text：prune-with-input
+
 import paddle.fluid as fluid
 import paddle.fluid.optimizer as optimizer
 import numpy as np