add size method for large scale

python/paddle/fluid/communicator.py

@@ -162,3 +162,6 @@ class LargeScaleKV(object):
 
     def load(self, varname, dirname):
         self.scale_kv.load(varname, dirname)
+
+    def size(self, varname):
+        return self.scale_kv.size(varname)

python/paddle/static/nn/__init__.py

@@ -34,6 +34,7 @@ __all__ = [
     'prelu',
     'row_conv',
     'spectral_norm',
+    'reshape',
 ]
 
 from ...fluid.layers import fc  #DEFINE_ALIAS
@@ -58,3 +59,4 @@ from ...fluid.layers import row_conv  #DEFINE_ALIAS
 from ...fluid.layers import spectral_norm  #DEFINE_ALIAS
 
 from ...fluid.input import embedding  #DEFINE_ALIAS
+from ...fluid.layers import reshape  #DEFINE_ALIAS