multihead_attention: set values from correct shape to tests

Author: OneAdder

test/test_multihead_attention_layer.f90

@@ -191,9 +191,10 @@ subroutine test_multihead_attention_backward(attention, ok)
     real :: input(3, 4, 1) = reshape([0.0, 10.1, 0.2, 10.3, 0.4, 10.5, 0.6, 10.7, 10.8, 0.9, 0.11, 0.12], [3, 4, 1])
     real :: gradient(3, 4, 1) = reshape([0.1, 3. , 2. , 0.1, 3. , 3. , 0.1, 2. , 0.1, 3. , 0.1, 3. ], [3, 4, 1])
     real :: expected_output_flat(12) = [&
-        0.489710003, 0.240968466, -3.35404873E-02, 0.489710003,&
-        0.240968466, -3.35404873E-02, 0.489710003, 0.240968466,&
-        -3.35404873E-02, 0.489710003, 0.240968466, -3.35404873E-02&
+        -2.29912549E-02, 0.381484956, 0.453185737,&
+        -2.29912549E-02, 0.381484956, 0.453185737,&
+        -2.29912549E-02, 0.381484956, 0.453185737,&
+        -2.29912549E-02, 0.381484956, 0.453185737&
     ]
     real :: expected_shape(3) = [3, 4, 1]
     real :: output(3, 4, 1)