@@ -3,6 +3,13 @@
33{-# LANGUAGE TypeFamilies #-}
44{-# LANGUAGE FlexibleInstances #-}
55{-# LANGUAGE MultiParamTypeClasses #-}
6+{-|
7+Module : Grenade.Core.Softmax
8+Description : Softmax loss layer
9+Copyright : (c) Huw Campbell, 2016-2017
10+License : BSD2
11+Stability : experimental
12+-}
613module Grenade.Layers.Softmax (
714 Softmax (..)
815 , softmax
@@ -17,8 +24,12 @@ import Grenade.Core
1724import Numeric.LinearAlgebra.Static as LAS
1825
1926-- | A Softmax layer
27+--
2028-- This layer is like a logit layer, but normalises
2129-- a set of matricies to be probabilities.
30+--
31+-- One can use this layer as the last layer in a network
32+-- if they need normalised probabilities.
2233data Softmax = Softmax
2334 deriving Show
2435