Skip to content

Commit 9d2f634

Browse files
committed
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

+2
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)