Skip to content

Commit 979334e

Browse files
committed
Fix other unittest
1 parent ae57b4e commit 979334e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

python/paddle/v2/tests/test_layer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@
2323
pixel = layer.data(name='pixel', type=data_type.dense_vector(128))
2424
label = layer.data(name='label', type=data_type.integer_value(10))
2525
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)