@@ -6,6 +6,13 @@
6{-# LANGUAGE TypeFamilies #-}
6{-# LANGUAGE TypeFamilies #-}
7{-# LANGUAGE MultiParamTypeClasses #-}
7{-# LANGUAGE MultiParamTypeClasses #-}
8{-# LANGUAGE FlexibleContexts #-}
8{-# LANGUAGE FlexibleContexts #-}
10+Module : Grenade.Core.Pooling
11+Description : Max Pooling layer for 2D and 3D images
12+Copyright : (c) Huw Campbell, 2016-2017
14+Stability : experimental
9module Grenade.Layers.Pooling (
16module Grenade.Layers.Pooling (
10 Pooling (..)
17 Pooling (..)
@@ -22,6 +29,7 @@ import Grenade.Layers.Internal.Pooling
22import Numeric.LinearAlgebra.Static as LAS hiding ((|||), build, toRows)
29import Numeric.LinearAlgebra.Static as LAS hiding ((|||), build, toRows)
24-- | A pooling layer for a neural network.
31-- | A pooling layer for a neural network.
25-- Does a max pooling, looking over a kernel similarly to the convolution network, but returning
33-- Does a max pooling, looking over a kernel similarly to the convolution network, but returning
26-- maxarg only. This layer is often used to provide minor amounts of translational invariance.
34-- maxarg only. This layer is often used to provide minor amounts of translational invariance.