-- Hoogle documentation, generated by Haddock
-- See Hoogle, http://www.haskell.org/hoogle/


-- | Deep Learning in Haskell
--   
--   Implements type-safe deep neural networks
@package deeplearning-hs
@version 0.1.0.0


module DeepLearning.ConvNet

-- | '(>->)' composes two forward activation functions
(>->) :: (Monad m, Shape sh, Shape sh', Shape sh'') => Forward m sh sh' -> Forward m sh' sh'' -> Forward m sh sh''

-- | Delayed activation matrix
type DVol sh = Array D sh Double

-- | The <a>Forward</a> function represents a single forward pass through a
--   layer.
type Forward m sh sh' = Vol sh -> WriterT [Vector Double] m (DVol sh')

-- | <a>InnerLayer</a> represents an inner layer of a neural network that
--   can accept backpropagation input from higher layers
class (Shape sh, Shape sh') => InnerLayer a sh sh' | a -> sh, a -> sh'

-- | <a>SoftMaxLayer</a> computes the softmax activation function.
data SoftMaxLayer
SoftMaxLayer :: SoftMaxLayer

-- | <a>TopLayer</a> is a top level layer that can initialize a
--   backpropagation pass.
class TopLayer a

-- | Activation matrix
type Vol sh = Array U sh Double

-- | <tt>FlowNetwork</tt> builds a network of the form
--   
--   <pre>
--   Input Layer              Output Softmax
--      +--+
--      |  |   Inner Layers    +--+   +--+
--      |  |                   |  |   |  |
--      |  |   +-+   +-+  +-+  |  |   |  |
--      |  +---+ +---+ +--+ +--+  +---&gt;  |
--      |  |   +-+   +-+  +-+  |  |   |  |
--      |  |                   |  |   |  |
--      |  |                   +--+   +--+
--      +--+
--   </pre>
flowNetwork :: (Monad m, Shape sh) => sh -> Int -> Int -> Int -> Forward m sh DIM1

-- | <a>net1</a> constructs a single-layer fully connected perceptron with
--   softmax output.
net1 :: (Monad m, InnerLayer a sh DIM1, TopLayer a1) => a -> a1 -> Forward m sh DIM1

-- | <a>net1</a> constructs a two-layer fully connected MLP with softmax
--   output.
net2 :: (Monad m, InnerLayer a sh sh', InnerLayer a1 sh' DIM1, TopLayer a2) => a -> a1 -> a2 -> Forward m sh DIM1

-- | <a>newFC</a> constructs a new fully connected layer
newFC :: Shape sh => sh -> Int -> FullyConnectedLayer sh

-- | <a>withActivations</a> computes the output activation, along with the
--   intermediate activations
withActivations :: Forward m sh sh' -> Vol sh -> m (DVol sh', [Vector Double])
instance Shape sh => InnerLayer (FullyConnectedLayer sh) sh DIM1
instance TopLayer SoftMaxLayer


module DeepLearning.Util

-- | Sample 3x3 matrix used for demonstrations and tests
testShape :: (Z :. Int) :. Int

-- | Random 3x3 matrix
testInput :: Shape sh => sh -> Array U sh Double

-- | Random single-layer network
testNet :: (Monad m, Shape sh) => sh -> Int -> Forward m sh DIM1