Change unittest's weight dimension to 1

ae57b4e7 · Yu Yang · a7730d40 · ae57b4e7
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/v2/tests/test_layer.py python/paddle/v2/tests/test_layer.py +1 -1

未找到文件。
--- a/python/paddle/v2/tests/test_layer.py
+++ b/python/paddle/v2/tests/test_layer.py
@@ -22,7 +22,7 @@ import paddle.v2.networks as networks

 pixel = layer.data(name='pixel', type=data_type.dense_vector(128))
 label = layer.data(name='label', type=data_type.integer_value(10))
-weight = layer.data(name='weight', type=data_type.dense_vector(10))
+weight = layer.data(name='weight', type=data_type.dense_vector(1))
 score = layer.data(name='score', type=data_type.dense_vector(1))

 hidden = layer.fc(input=pixel,