Skip to content

Commit 73487bd

Browse files
committed
Getting output from RNN
1 parent bc96589 commit 73487bd

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/nf/nf_layer_submodule.f90

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,8 @@ pure module subroutine get_output_1d(self, output)
228228
allocate(output, source=this_layer % output)
229229
type is(flatten_layer)
230230
allocate(output, source=this_layer % output)
231+
type is(rnn_layer)
232+
allocate(output, source=this_layer % output)
231233
class default
232234
error stop '1-d output can only be read from an input1d, dense, or flatten layer.'
233235

0 commit comments

Comments
 (0)