Skip to content

Commit 53090e3

Browse files
authored
Merge pull request #1650 from reyoung/feature/fix_unittest_error
Change unittest's weight dimension to 1
2 parents 7f42793 + 979334e commit 53090e3

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

python/paddle/v2/tests/test_layer.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,9 @@
2222

2323
pixel = layer.data(name='pixel', type=data_type.dense_vector(128))
2424
label = layer.data(name='label', type=data_type.integer_value(10))
25-
weight = layer.data(name='weight', type=data_type.dense_vector(10))
25+
weight = layer.data(name='weight', type=data_type.dense_vector(1))
26+
combine_weight = layer.data(
27+
name='weight_combine', type=data_type.dense_vector(10))
2628
score = layer.data(name='score', type=data_type.dense_vector(1))
2729

2830
hidden = layer.fc(input=pixel,
@@ -81,7 +83,8 @@ def test_aggregate_layer(self):
8183
class MathLayerTest(unittest.TestCase):
8284
def test_math_layer(self):
8385
addto = layer.addto(input=[pixel, pixel])
84-
linear_comb = layer.linear_comb(weights=weight, vectors=hidden, size=10)
86+
linear_comb = layer.linear_comb(
87+
weights=combine_weight, vectors=hidden, size=10)
8588
interpolation = layer.interpolation(
8689
input=[hidden, hidden], weight=score)
8790
bilinear = layer.bilinear_interp(input=conv, out_size_x=4, out_size_y=4)

0 commit comments

Comments
 (0)