Skip to content

Commit 33b9c5d

Browse files
committed
Enable get_output() for maxpool2d layers
1 parent 437a53a commit 33b9c5d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/nf_layer_submodule.f90

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,10 @@ pure module subroutine get_output_3d(self, output)
9191
allocate(output, source=this_layer % output)
9292
type is(conv2d_layer)
9393
allocate(output, source=this_layer % output)
94+
type is(maxpool2d_layer)
95+
allocate(output, source=this_layer % output)
9496
class default
95-
error stop '3-d output can only be read from an input3d or conv2d layer.'
97+
error stop '3-d output can only be read from an input3d, conv2d, or maxpool2d layer.'
9698

9799
end select
98100

0 commit comments

Comments
 (0)