Skip to content

Commit 3396edb

Browse files
committed
Enable network % output() of rnn_layer
1 parent 736ded7 commit 3396edb

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/nf/nf_network_submodule.f90

+3
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
use nf_loss, only: quadratic_derivative
1616
use nf_optimizers, only: optimizer_base_type, sgd
1717
use nf_parallel, only: tile_indices
18+
use nf_rnn_layer, only: rnn_layer
1819
use nf_activation, only: activation_function, &
1920
elu, &
2021
exponential, &
@@ -391,6 +392,8 @@ module function predict_1d(self, input) result(res)
391392
res = output_layer % output
392393
type is(flatten_layer)
393394
res = output_layer % output
395+
type is(rnn_layer)
396+
res = output_layer % output
394397
class default
395398
error stop 'network % output not implemented for this output layer'
396399
end select

0 commit comments

Comments
 (0)