From 8461ab17f087b41b74d66f3bbbe7ac2d24e29d59 Mon Sep 17 00:00:00 2001
From: LielinJiang <50691816+LielinJiang@users.noreply.github.com>
Date: Thu, 10 Jun 2021 19:12:34 +0800
Subject: [PATCH] add sample code for summary (#33337) (#33427)

---
 python/paddle/hapi/model_summary.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/python/paddle/hapi/model_summary.py b/python/paddle/hapi/model_summary.py
index d78196d944..93f1a5a37a 100644
--- a/python/paddle/hapi/model_summary.py
+++ b/python/paddle/hapi/model_summary.py
@@ -80,6 +80,23 @@ def summary(net, input_size, dtypes=None):
             params_info = paddle.summary(lenet, (1, 1, 28, 28))
             print(params_info)
 
+            # multi input demo
+            class LeNetMultiInput(LeNet):
+
+                def forward(self, inputs, y):
+                    x = self.features(inputs)
+
+                    if self.num_classes > 0:
+                        x = paddle.flatten(x, 1)
+                        x = self.fc(x + y)
+                    return x
+            
+            lenet_multi_input = LeNetMultiInput()
+
+            params_info = paddle.summary(lenet_multi_input, [(1, 1, 28, 28), (1, 400)], 
+                                        ['float32', 'float32'])
+            print(params_info)
+
     """
     if isinstance(input_size, InputSpec):
         _input_size = tuple(input_size.shape)
-- 
GitLab