We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5e054fa commit d66f9ffCopy full SHA for d66f9ff
src/pooling.jl
@@ -141,14 +141,14 @@ expand(N, i::Tuple) = i
141
expand(N, i::Integer) = ntuple(_ -> i, N)
142
143
function maxpool(x, k::NTuple{N, Integer}; pad = 0, stride = k) where N
144
- pad = expand(Val(2*N), pad)
+ pad = expand(Val(N), pad)
145
stride = expand(Val(N), stride)
146
pdims = PoolDims(x, k; padding = pad, stride = stride)
147
return maxpool(x, pdims)
148
end
149
150
function meanpool(x, k::NTuple{N, Integer}; pad = 0, stride = k) where N
151
152
153
154
return meanpool(x, pdims)
0 commit comments