Replace `FC` with `Linear`

mnist.py

@@ -19,7 +19,7 @@ import numpy as np
 import paddle
 from paddle import fluid
 from paddle.fluid.optimizer import MomentumOptimizer
-from paddle.fluid.dygraph.nn import Conv2D, Pool2D, FC
+from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear
 
 from model import Model, shape_hints
 
@@ -83,7 +83,7 @@ class MNIST(Model):
         pool_2_shape = 50 * 4 * 4
         SIZE = 10
         scale = (2.0 / (pool_2_shape**2 * SIZE))**0.5
-        self._fc = FC(self.full_name(),
+        self._fc = Linear(800,
                           10,
                           param_attr=fluid.param_attr.ParamAttr(
                               initializer=fluid.initializer.NormalInitializer(
@@ -95,9 +95,12 @@ class MNIST(Model):
         if self.mode == 'test':  # XXX demo purpose
             x = self._simple_img_conv_pool_1(inputs)
             x = self._simple_img_conv_pool_2(x)
+            x = fluid.layers.flatten(x, axis=1)
+            x = self._fc(x)
         else:
             x = self._simple_img_conv_pool_1(inputs)
             x = self._simple_img_conv_pool_2(x)
+            x = fluid.layers.flatten(x, axis=1)
             x = self._fc(x)
         return x