|
22 | 22 |
|
23 | 23 | pixel = layer.data(name='pixel', type=data_type.dense_vector(128))
|
24 | 24 | 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)) |
26 | 28 | score = layer.data(name='score', type=data_type.dense_vector(1))
|
27 | 29 |
|
28 | 30 | hidden = layer.fc(input=pixel,
|
@@ -81,7 +83,8 @@ def test_aggregate_layer(self):
|
81 | 83 | class MathLayerTest(unittest.TestCase):
|
82 | 84 | def test_math_layer(self):
|
83 | 85 | 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) |
85 | 88 | interpolation = layer.interpolation(
|
86 | 89 | input=[hidden, hidden], weight=score)
|
87 | 90 | bilinear = layer.bilinear_interp(input=conv, out_size_x=4, out_size_y=4)
|
|
0 commit comments