MXNet.NN.Layer

Documentation

variable :: String -> IO SymbolHandle

convolution :: (MatchKVList kvs '["stride" := String, "dilate" := String, "pad" := String, "num_group" := Int, "workspace" := Int, "no_bias" := Bool, "cudnn_tune" := String, "cudnn_off" := Bool, "layout" := String], ShowKV kvs) => String -> SymbolHandle -> [Int] -> Int -> HMap kvs -> IO SymbolHandle

fullyConnected :: (MatchKVList kvs '["no_bias" := Bool, "flatten" := Bool], ShowKV kvs) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle

data PoolingMethod

Constructors

PoolingMax

PoolingAvg

PoolingSum

poolingMethodToStr :: PoolingMethod -> String

pooling :: (MatchKVList kvs '["global_pool" := Bool, "cudnn_off" := Bool, "pooling_convention" := String, "stride" := String, "pad" := String], ShowKV kvs) => String -> SymbolHandle -> [Int] -> PoolingMethod -> HMap kvs -> IO SymbolHandle

flatten :: String -> SymbolHandle -> IO SymbolHandle

data ActivationType

Constructors

Relu

Sigmoid

Tanh

SoftRelu

activationTypeToStr :: ActivationType -> String

activation :: String -> SymbolHandle -> ActivationType -> IO SymbolHandle

softmaxoutput :: (MatchKVList kvs '["grad_scale" := Float, "ignore_label" := Float, "multi_output" := Bool, "use_ignore" := Bool, "preserve_shape" := Bool, "normalization" := String, "out_grad" := Bool, "smooth_alpha" := Float], ShowKV kvs) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle