From 3396edb6579ec4ebd5f2e406a3f8f4f62209f6aa Mon Sep 17 00:00:00 2001
From: milancurcic <caomaco@gmail.com>
Date: Wed, 6 Dec 2023 11:36:55 -0500
Subject: [PATCH] Enable network % output() of rnn_layer

---
 src/nf/nf_network_submodule.f90 | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/nf/nf_network_submodule.f90 b/src/nf/nf_network_submodule.f90
index e6168857..c0af3608 100644
--- 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