@@ -9,7 +9,11 @@ fn local_response_normalization() {
9
9
let batches = 1 ;
10
10
let width_height: usize = 3 ;
11
11
let channels: usize = 4 ;
12
- let data: Vec < f32 > = [ 1. , 1. , 2. , 4. , 2. , 2. , 1. , 2. , 3. , 1. , 2. , 1. , 4. , 2. , 3. , 5. , 3. , 3. , 2. , 2. , 6. , 2. , 3. , 1. , 7. , 3. , 4. , 2. , 8. , 4. , 3. , 2. , 9. , 3. , 4. , 4. ] . to_vec ( ) ;
12
+ let data: Vec < f32 > = [
13
+ 1. , 1. , 2. , 4. , 2. , 2. , 1. , 2. , 3. , 1. , 2. , 1. , 4. , 2. , 3. , 5. , 3. , 3. , 2. , 2. , 6. , 2. , 3. ,
14
+ 1. , 7. , 3. , 4. , 2. , 8. , 4. , 3. , 2. , 9. , 3. , 4. , 4. ,
15
+ ]
16
+ . to_vec ( ) ;
13
17
14
18
let shape = vec ! [
15
19
batches as i64 ,
@@ -20,21 +24,22 @@ fn local_response_normalization() {
20
24
input_data. insert ( "X" . to_string ( ) , data. as_slice ( ) . into ( ) ) ;
21
25
22
26
let bn_model = model ( graph (
23
- vec ! [ tensor( "X" , & shape) ] , // input
24
- vec ! [ tensor( "Y" , & shape) ] , // output
25
- vec ! [ ] , // infos
26
- vec ! [ ] , // intializers
27
-
27
+ vec ! [ tensor( "X" , & shape) ] , // input
28
+ vec ! [ tensor( "Y" , & shape) ] , // output
29
+ vec ! [ ] , // infos
30
+ vec ! [ ] , // intializers
28
31
// nodes
29
32
vec ! [ node(
30
33
vec![ "X" ] ,
31
34
vec![ "Y" ] ,
32
35
"lrn" ,
33
36
"LRN" ,
34
- vec![ attribute( "alpha" , 1.0 ) ,
35
- attribute( "beta" , 1.0 ) ,
36
- attribute( "bias" , 0.0 ) ,
37
- attribute( "size" , 2 ) ] ,
37
+ vec![
38
+ attribute( "alpha" , 1.0 ) ,
39
+ attribute( "beta" , 1.0 ) ,
40
+ attribute( "bias" , 0.0 ) ,
41
+ attribute( "size" , 2 ) ,
42
+ ] ,
38
43
) ] ,
39
44
) ) ;
40
45
@@ -47,6 +52,10 @@ fn local_response_normalization() {
47
52
48
53
common:: assert_eq_vector (
49
54
out_y. try_into ( ) . unwrap ( ) ,
50
- & [ 1.0 , 0.4 , 0.2 , 0.5 , 0.5 , 0.8 , 0.4 , 1.0 , 0.6 , 0.4 , 0.8 , 2.0 , 0.4 , 0.30769232 , 0.1764706 , 0.39999998 , 0.33333334 , 0.4615385 , 0.5 , 1.0 , 0.3 , 0.30769232 , 0.6 , 2.0 , 0.2413793 , 0.24 , 0.4 , 1.0 , 0.2 , 0.32 , 0.4615385 , 1.0 , 0.2 , 0.24 , 0.25 , 0.5 ] ,
55
+ & [
56
+ 1.0 , 0.4 , 0.2 , 0.5 , 0.5 , 0.8 , 0.4 , 1.0 , 0.6 , 0.4 , 0.8 , 2.0 , 0.4 , 0.30769232 , 0.1764706 ,
57
+ 0.39999998 , 0.33333334 , 0.4615385 , 0.5 , 1.0 , 0.3 , 0.30769232 , 0.6 , 2.0 , 0.2413793 ,
58
+ 0.24 , 0.4 , 1.0 , 0.2 , 0.32 , 0.4615385 , 1.0 , 0.2 , 0.24 , 0.25 , 0.5 ,
59
+ ] ,
51
60
) ;
52
- }
61
+ }
0 commit comments