maxpool and meanpool added

Author: thebhatman

src/pooling.jl

@@ -114,7 +114,7 @@ for backend in (Symbol(), :_direct, :_im2col)
                 fill!(y, xT(0))
                 return $(Symbol("$(name)$(backend)!"))(y, x, pdims; kwargs...)
             end
-            
+
             # Backprops too
             @timeit_debug to function $(Symbol("∇$(name)$(backend)"))(
                             dy::AbstractArray{T,N}, y::AbstractArray{T,N},
@@ -136,3 +136,20 @@ if is_nnpack_available()
         return func(x, pdims; kwargs...)
     end
 end
+
+expand(N, i::Tuple) = i
+expand(N, i::Integer) = ntuple(_ -> i, N)
+
+function maxpool(x, k::NTuple{N, Integer}; pad = 0, stride = k) where N
+    pad = expand(Val(2*N), pad)
+    stride = expand(Val(N), stride)
+    pdims = PoolDims(x, k; padding = pad, stride = stride)
+    return maxpool(x, pdims)
+end
+
+function meanpool(x, k::NTuple{N, Integer}; pad = 0, stride = k) where N
+    pad = expand(Val(2*N), pad)
+    stride = expand(Val(N), stride)
+    pdims = PoolDims(x, k; padding = pad, stride = stride)
+    return meanpool(x, pdims)
+end