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