Skip to content

Commit d66f9ff

Browse files
authored
Changed 2*N to N.
1 parent 5e054fa commit d66f9ff

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/pooling.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,14 +141,14 @@ expand(N, i::Tuple) = i
141141
expand(N, i::Integer) = ntuple(_ -> i, N)
142142

143143
function maxpool(x, k::NTuple{N, Integer}; pad = 0, stride = k) where N
144-
pad = expand(Val(2*N), pad)
144+
pad = expand(Val(N), pad)
145145
stride = expand(Val(N), stride)
146146
pdims = PoolDims(x, k; padding = pad, stride = stride)
147147
return maxpool(x, pdims)
148148
end
149149

150150
function meanpool(x, k::NTuple{N, Integer}; pad = 0, stride = k) where N
151-
pad = expand(Val(2*N), pad)
151+
pad = expand(Val(N), pad)
152152
stride = expand(Val(N), stride)
153153
pdims = PoolDims(x, k; padding = pad, stride = stride)
154154
return meanpool(x, pdims)

0 commit comments

Comments
 (0)