Enable network % output() of rnn_layer

modern-fortran · Dec 6, 2023 · 3396edb · 3396edb
1 parent 736ded7
commit 3396edb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/nf/nf_network_submodule.f90 b/src/nf/nf_network_submodule.f90
@@ -15,6 +15,7 @@
   use nf_loss, only: quadratic_derivative
   use nf_optimizers, only: optimizer_base_type, sgd
   use nf_parallel, only: tile_indices
+  use nf_rnn_layer, only: rnn_layer
   use nf_activation, only: activation_function, &
                            elu, &
                            exponential, &
@@ -391,6 +392,8 @@ module function predict_1d(self, input) result(res)
         res = output_layer % output
       type is(flatten_layer)
         res = output_layer % output
+      type is(rnn_layer)
+        res = output_layer % output
       class default
         error stop 'network % output not implemented for this output layer'
     end select