Skip to content

Commit ee7b598

Browse files
authored
Update nf_layer_submodule.f90
bug fix
1 parent cf2caf6 commit ee7b598

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/nf/nf_layer_submodule.f90

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,13 @@ pure module subroutine backward_2d(self, previous, gradient)
8787
type is(locally_connected_1d_layer)
8888
call this_layer % backward(prev_layer % output, gradient)
8989
end select
90+
91+
type is(reshape2d_layer)
92+
93+
select type(prev_layer => previous % p)
94+
type is(input2d_layer)
95+
call this_layer % backward(prev_layer % output, gradient)
96+
end select
9097

9198
end select
9299

0 commit comments

Comments
 (0)