Skip to content

Commit 3a8cecc

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

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/nf/nf_layer_submodule.f90

+8
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,14 @@ pure module subroutine forward(self, input)
256256
call this_layer % forward(prev_layer % output)
257257
end select
258258

259+
type is(reshape2d_layer)
260+
261+
! Upstream layers permitted: input1d, dense, flatten
262+
select type(prev_layer => input % p)
263+
type is(input2d_layer)
264+
call this_layer % forward(prev_layer % output)
265+
end select
266+
259267
end select
260268

261269
end subroutine forward

0 commit comments

Comments
 (0)