MXNet.NN.Types

Documentation

data Parameter a

A parameter is two NDArray to back a Symbol

Constructors

Parameter

Fields

• _param_in :: NDArray a
• _param_grad :: NDArray a

Instances

(Pretty a, DType a) => Show (Parameter a)

Methods

showsPrec :: Int -> Parameter a -> ShowS

show :: Parameter a -> String

showList :: [Parameter a] -> ShowS

data Session a

Session is all the Parameters and a Context type Session a = (M.HashMap String (Parameter a), Context)

Constructors

Session

Fields

• _sess_param :: !(HashMap String (Parameter a))
• _sess_context :: !Context

sess_param :: forall a a. Lens (Session a) (Session a) (HashMap String (Parameter a)) (HashMap String (Parameter a))

sess_context :: forall a. Lens' (Session a) Context

type TrainM a m = StateT (Session a) m

TrainM is a StateT monad

type Initializer a = Context -> [Int] -> IO (NDArray a)

Initializer is about how to create a NDArray from a given shape.

Usually, it can be a wrapper of MXNet operators, such as random_uniform, random_normal, random_gamma, etc..