Fix dims in config parser for factorization machine layer

python/paddle/trainer/config_parser.py

@@ -3794,7 +3794,7 @@ class FactorizationMachineLayer(LayerBase):
         self.config.factor_size = factor_size
         input_layer = self.get_input_layer(0)
         psize = input_layer.size * factor_size
-        dims = [input_layer.size, 1]
+        dims = [input_layer.size, factor_size]
         self.create_input_parameter(0, psize, dims)
 
 

python/paddle/trainer_config_helpers/tests/configs/protostr/test_factorization_machine.protostr

@@ -22,7 +22,7 @@ parameters {
   initial_mean: 0.0
   initial_std: 0.03125
   dims: 1024
-  dims: 1
+  dims: 10
   initial_strategy: 0
   initial_smart: true
 }