feat: get_params()

Author: castelao

src/nf/nf_layer_submodule.f90

@@ -320,6 +320,8 @@ module function get_params(self) result(params)
         ! No parameters to get.
       type is (reshape3d_layer)
         ! No parameters to get.
+      type is (rnn_layer)
+        params = this_layer % get_params()
       class default
         error stop 'Unknown layer type.'
     end select

src/nf/nf_rnn_layer.f90

@@ -37,7 +37,7 @@ module nf_rnn_layer
     !procedure :: forward
     !procedure :: get_gradients
     procedure :: get_num_params
-    !procedure :: get_params
+    procedure :: get_params
     procedure :: init
     !procedure :: set_params
 

src/nf/nf_rnn_layer_submodule.f90

@@ -63,6 +63,7 @@ pure module function get_params(self) result(params)
 
     params = [ &
       pack(self % weights, .true.), &
+      pack(self % recurrent, .true.), &
       pack(self % biases, .true.) &
     ]