Add unit test for dynamic_flops with multiple outputs

Author: yingyibiao

python/paddle/tests/test_model.py

@@ -733,30 +733,14 @@ def customize_dropout(m, x, y):
             print_detail=True)
 
     def test_dynamic_flops_with_multiple_outputs(self):
-        class MyMnist(paddle.nn.Layer):
-            def __init__(self):
-                super(MyMnist, self).__init__()
-                self.flatten = paddle.nn.Flatten()
-                self.linear_1 = paddle.nn.Linear(784, 512)
-                self.linear_2 = paddle.nn.Linear(512, 10)
-                self.relu = paddle.nn.ReLU()
-                self.dropout = paddle.nn.Dropout(0.2)
-
-            def forward(self, inputs):
-                y = self.flatten(inputs)
-                y = self.linear_1(y)
-                y1 = self.relu(y)
-                y2 = self.dropout(y1)
-                y2 = self.linear_2(y2)
-                return y1, y2
-
-        net = MyMnist()
+        net = paddle.nn.MaxPool2D(
+            kernel_size=2, stride=2, padding=0, return_mask=True)
 
         def customize_dropout(m, x, y):
             m.total_ops += 0
 
         paddle.flops(
-            net, [1, 1, 28, 28],
+            net, [1, 2, 32, 32],
             custom_ops={paddle.nn.Dropout: customize_dropout},
             print_detail=True)