diff --git a/python/paddle/hapi/model_summary.py b/python/paddle/hapi/model_summary.py
index babbe962a952528f462ae4e879a0d47c69550755..9f2769e1ca285766dcbbd6d852ef5540574eed1c 100644
--- a/python/paddle/hapi/model_summary.py
+++ b/python/paddle/hapi/model_summary.py
@@ -341,10 +341,12 @@ def summary_string(model, input_size, dtypes=None):
         total_params += summary[layer]["nb_params"]
 
         try:
-            total_output += np.prod(summary[layer]["output_shape"])
+            total_output += np.sum(
+                np.prod(
+                    summary[layer]["output_shape"], axis=-1))
         except:
             for output_shape in summary[layer]["output_shape"]:
-                total_output += np.prod(output_shape)
+                total_output += np.sum(np.prod(output_shape, axis=-1))
 
         if "trainable" in summary[layer]:
             if summary[layer]["trainable"] == True: