Skip to content

Commit 663860d

Browse files
committed
fix layer constructor
1 parent b44a151 commit 663860d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/nf/nf_layer_constructors.f90

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,8 @@ module function locally_connected2d(filters, kernel_size, activation) result(res
179179
!! Width of the convolution window, commonly 3 or 5
180180
class(activation_function), intent(in), optional :: activation
181181
!! Activation function (default sigmoid)
182+
integer, intent(in), optional :: stride
183+
!! Size of the stride (default 1)
182184
type(layer) :: res
183185
!! Resulting layer instance
184186
end function locally_connected2d

0 commit comments

Comments
 (0)