diff --git a/python/paddle/nn/quant/quant_layers.py b/python/paddle/nn/quant/quant_layers.py
index 040b04f5e7bf1efc1dcda5c299a9f9cd46a7f28b..220e0c34a1e53cc9e578d53de6b53d298efd4760 100644
--- a/python/paddle/nn/quant/quant_layers.py
+++ b/python/paddle/nn/quant/quant_layers.py
@@ -677,7 +677,8 @@ class MAOutputScaleLayer(layers.Layer):
     def forward(self, *inputs, **kwargs):
         out = self._layer(*inputs, **kwargs)
         # TODO (jc): support the ops of several outputs
-        if (isinstance(out, list) or isinstance(out, tuple)) and len(out) > 1:
+        if (isinstance(out, list) or isinstance(out, tuple) or
+                isinstance(out, dict)):
             return out
         else:
             return self._ma_output_scale(out)