feat: set_params()

Author: castelao

src/nf/nf_rnn_layer.f90

@@ -39,7 +39,7 @@ module nf_rnn_layer
     procedure :: get_num_params
     procedure :: get_params
     procedure :: init
-    !procedure :: set_params
+    procedure :: set_params
 
   end type rnn_layer
 

src/nf/nf_rnn_layer_submodule.f90

@@ -85,21 +85,32 @@ end function get_gradients
   module subroutine set_params(self, params)
     class(rnn_layer), intent(in out) :: self
     real, intent(in) :: params(:)
+    integer :: first, last
 
     ! check if the number of parameters is correct
     if (size(params) /= self % get_num_params()) then
       error stop 'Error: number of parameters does not match'
     end if
 
     ! reshape the weights
+    last = self % input_size * self % output_size
     self % weights = reshape( &
-      params(:self % input_size * self % output_size), &
+      params(:last), &
       [self % input_size, self % output_size] &
     )
 
+    ! reshape the recurrent weights
+    first = last + 1
+    last = first + self % output_size * self % output_size
+    self % recurrent = reshape( &
+      params(first:last), &
+      [self % output_size, self % output_size] &
+    )
+
     ! reshape the biases
+    first = last + 1
     self % biases = reshape( &
-      params(self % input_size * self % output_size + 1:), &
+      params(first:), &
       [self % output_size] &
     )