Skip to content

Commit 0760191

Browse files
committed
fix API constructor layer
1 parent b9f418f commit 0760191

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/nf/nf_layer_constructors.f90

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ end function conv2d
156156

157157
interface locally_connected
158158

159-
module function locally_connected2d(filters, kernel_size, activation) result(res)
159+
module function locally_connected2d(filters, kernel_size, activation, stride) result(res)
160160
!! 1-d locally connected network constructor
161161
!!
162162
!! This layer is for building 1-d locally connected network.

0 commit comments

Comments
 (0)