Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 9d2f634

Browse files
committedNov 14, 2023
feat: layer getting gradient from RNN
1 parent 1bef0aa commit 9d2f634

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
@@ -379,6 +379,8 @@ pure module function get_gradients(self) result(gradients)
379379
! No gradients to get.
380380
type is (reshape3d_layer)
381381
! No gradients to get.
382+
type is (rnn_layer)
383+
gradients = this_layer % get_gradients()
382384
class default
383385
error stop 'get_gradients() with unknown layer type.'
384386
end select

0 commit comments

Comments
 (0)
Please sign in to comment.