File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change 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
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 )
You can’t perform that action at this time.
0 commit comments