tensor-ops-0.1.0.0: Tensor manipulation operations with tensorflow-style automatic differentiation

Safe HaskellNone
LanguageHaskell2010

TensorOps.Learn.NeuralNet

Synopsis

Documentation

newtype Activation k Source #

Constructors

Act 

Fields

actMap :: (forall a. RealFloat a => a -> a) -> Activation k Source #

actMap' :: (forall a. RealFloat a => a -> a) -> (forall a. RealFloat a => a -> a) -> Activation k Source #

logistic :: Floating a => a -> a Source #

logistic' :: Floating a => a -> a Source #

softmax :: SingI i => TOp '['[i]] '['[i]] Source #

squaredError :: forall o. SingI o => TOp '['[o], '[o]] '['[]] Source #

crossEntropy :: forall o. SingI o => TOp '['[o], '[o]] '['[]] Source #

Second item in stack is the "target"