MXNet.Core.Base

Interfaces in core module of MXNet.

Necessary raw functions

mxGetLastError :: IO String

Handle size_t type.

Get the string message of last error.

mxListAllOpNames :: IO (Int, [String])

List all the available operator names, include entries.

NDArray

data NDArray a

NDArray type alias.

Instances

Neural NDArray

Methods

fullyConnected :: DType a => NDArray a -> NDArray a -> NDArray a -> Int -> IO (NDArray a)

correlation :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

activation :: DType a => NDArray a -> String -> IO (NDArray a)

leakyReLU :: DType a => NDArray a -> String -> IO (NDArray a)

softmaxActivation :: DType a => NDArray a -> IO (NDArray a)

dropout :: DType a => NDArray a -> Float -> IO (NDArray a)

batchNorm :: DType a => NDArray a -> NDArray a -> NDArray a -> NDArray a -> NDArray a -> IO (NDArray a)

instanceNorm :: DType a => NDArray a -> NDArray a -> NDArray a -> Float -> IO (NDArray a)

l2Normalization :: DType a => NDArray a -> Float -> String -> IO (NDArray a)

convolution :: DType a => NDArray a -> NDArray a -> NDArray a -> String -> Int -> IO (NDArray a)

lrn :: DType a => NDArray a -> Float -> Float -> Float -> Int -> IO (NDArray a)

deconvolution :: DType a => NDArray a -> NDArray a -> NDArray a -> String -> Int -> IO (NDArray a)

pooling :: DType a => NDArray a -> String -> String -> IO (NDArray a)

roiPooling :: DType a => NDArray a -> NDArray a -> String -> Int -> IO (NDArray a)

rnn :: DType a => NDArray a -> NDArray a -> NDArray a -> NDArray a -> Int -> Int -> String -> IO (NDArray a)

embedding :: DType a => NDArray a -> NDArray a -> Int -> Int -> IO (NDArray a)

bilinearSampler :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

gridGenerator :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

upSampling :: DType a => [NDArray a] -> Int -> String -> IO (NDArray a)

spatialTransformer :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

linearRegressionOutput :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

logisticRegressionOutput :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

softmaxOutput :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

maeRegressionOutput :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

svmOutput :: DType a => NDArray a -> NDArray a -> Int -> Float -> Bool -> IO (NDArray a)

softmaxCrossEntropy :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

smoothL1 :: DType a => NDArray a -> Float -> IO (NDArray a)

identityAttachKLSparsereg :: DType a => NDArray a -> IO (NDArray a)

makeLoss :: DType a => NDArray a -> Float -> Float -> String -> IO (NDArray a)

blockGrad :: DType a => NDArray a -> IO (NDArray a)

custom :: DType a => [NDArray a] -> String -> IO (NDArray a)

Tensor NDArray

Methods

dot :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

reshape :: DType a => NDArray a -> [Int] -> IO (NDArray a)

transpose :: DType a => NDArray a -> IO (NDArray a)

(+.) :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

(-.) :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

(*.) :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

(/.) :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

(^.) :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

(.+) :: DType a => NDArray a -> a -> IO (NDArray a)

(.-) :: DType a => NDArray a -> a -> IO (NDArray a)

(.*) :: DType a => NDArray a -> a -> IO (NDArray a)

(./) :: DType a => NDArray a -> a -> IO (NDArray a)

(.^) :: DType a => NDArray a -> a -> IO (NDArray a)

(..-) :: DType a => a -> NDArray a -> IO (NDArray a)

(../) :: DType a => a -> NDArray a -> IO (NDArray a)

(..^) :: DType a => a -> NDArray a -> IO (NDArray a)

(.+=) :: DType a => NDArray a -> a -> IO ()

(.-=) :: DType a => NDArray a -> a -> IO ()

(.*=) :: DType a => NDArray a -> a -> IO ()

(./=) :: DType a => NDArray a -> a -> IO ()

(.^=) :: DType a => NDArray a -> a -> IO ()

_Maximum :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

_Minimum :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

equal :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

notEqual :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

greater :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

greaterEqual :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

lesser :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

lesserEqual :: DType a => NDArray a -> NDArray a -> IO (NDArray a)

_Maximum' :: DType a => NDArray a -> a -> IO (NDArray a)

_Minimum' :: DType a => NDArray a -> a -> IO (NDArray a)

equal' :: DType a => NDArray a -> a -> IO (NDArray a)

notEqual' :: DType a => NDArray a -> a -> IO (NDArray a)

greater' :: DType a => NDArray a -> a -> IO (NDArray a)

greaterEqual' :: DType a => NDArray a -> a -> IO (NDArray a)

lesser' :: DType a => NDArray a -> a -> IO (NDArray a)

lesserEqual' :: DType a => NDArray a -> a -> IO (NDArray a)

(DType a, (~) * a Int8) => Eq (NDArray Int8)

Methods

(==) :: NDArray Int8 -> NDArray Int8 -> Bool

(/=) :: NDArray Int8 -> NDArray Int8 -> Bool

(DType a, (~) * a Int32) => Eq (NDArray Int32)

Methods

(==) :: NDArray Int32 -> NDArray Int32 -> Bool

(/=) :: NDArray Int32 -> NDArray Int32 -> Bool

(DType a, Floating a) => Eq (NDArray a)

Methods

(==) :: NDArray a -> NDArray a -> Bool

(/=) :: NDArray a -> NDArray a -> Bool

DType a => Floating (NDArray a)

Methods

pi :: NDArray a

exp :: NDArray a -> NDArray a

log :: NDArray a -> NDArray a

sqrt :: NDArray a -> NDArray a

(**) :: NDArray a -> NDArray a -> NDArray a

logBase :: NDArray a -> NDArray a -> NDArray a

sin :: NDArray a -> NDArray a

cos :: NDArray a -> NDArray a

tan :: NDArray a -> NDArray a

asin :: NDArray a -> NDArray a

acos :: NDArray a -> NDArray a

atan :: NDArray a -> NDArray a

sinh :: NDArray a -> NDArray a

cosh :: NDArray a -> NDArray a

tanh :: NDArray a -> NDArray a

asinh :: NDArray a -> NDArray a

acosh :: NDArray a -> NDArray a

atanh :: NDArray a -> NDArray a

log1p :: NDArray a -> NDArray a

expm1 :: NDArray a -> NDArray a

log1pexp :: NDArray a -> NDArray a

log1mexp :: NDArray a -> NDArray a

DType a => Fractional (NDArray a)

Methods

(/) :: NDArray a -> NDArray a -> NDArray a

recip :: NDArray a -> NDArray a

fromRational :: Rational -> NDArray a

DType a => Num (NDArray a)

Methods

(+) :: NDArray a -> NDArray a -> NDArray a

(-) :: NDArray a -> NDArray a -> NDArray a

(*) :: NDArray a -> NDArray a -> NDArray a

negate :: NDArray a -> NDArray a

abs :: NDArray a -> NDArray a

signum :: NDArray a -> NDArray a

fromInteger :: Integer -> NDArray a

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

Methods

showsPrec :: Int -> NDArray a -> ShowS

show :: NDArray a -> String

showList :: [NDArray a] -> ShowS

waitAll :: IO ()

Wait all async operation to finish in MXNet.

makeEmptyNDArray

Arguments

:: DType a

=> [Int]

Shape.

-> Context

Context/

-> Bool

If delayed allocate.

-> IO (NDArray a)

Make a new empty ndarray with specified shape, context and data type.

makeNDArray

Arguments

:: DType a

=> [Int]

size of every dimensions.

-> Context

-> Vector a

-> IO (NDArray a)

Make a new NDArray with given shape.

ndshape

Arguments

:: DType a

=> NDArray a

-> IO (Int, [Int])

Dimensions and size of every dimensions.

Get the shape of given NDArray.

ndsize

Arguments

:: DType a

=> NDArray a

-> IO Int

Dimensions and size of every dimensions.

Get size of the given ndarray.

context :: DType a => NDArray a -> IO Context

Get context of the given ndarray.

at

Arguments

:: DType a

=> NDArray a

-> Int

The index.

-> NDArray a

Return a sub ndarray that shares memory with current one.

items :: DType a => NDArray a -> IO (Vector a)

Get data stored in NDArray.

slice

Arguments

:: DType a

=> NDArray a

-> Int

The beginning index of slice.

-> Int

The end index of slices.

-> NDArray a

Return a sliced ndarray that shares memory with current one.

waitToRead :: DType a => NDArray a -> IO ()

Block until all pending writes operations on current ndarray are finished.

onehotEncode

Arguments

:: DType a

=> NDArray a

An ndarray containing indices of the categorical features.

-> NDArray a

The result holder of the encoding.

-> IO (NDArray a)

The encoding ndarray.

One hot encoding indices into matrix out.

zeros

Arguments

:: DType a

=> [Int]

Shape.

-> IO (NDArray a)

Create a new NDArray filled with 0, with specified shape and context.

ones

Arguments

:: DType a

=> [Int]

Shape.

-> IO (NDArray a)

Create a new NDArray filled with 1, with specified shape and context.

full

Arguments

:: DType a

=> [Int]

Shape.

-> a

Given value to fill the ndarray.

-> IO (NDArray a)

Create a new NDArray filled with given value, with specified shape and context.

array

Arguments

:: DType a

=> [Int]

Shape.

-> Vector a

-> IO (NDArray a)

Create a new NDArray that copies content from source_array.

Symbol

data Symbol a

Type alias for variable.

Instances

Neural Symbol

Methods

fullyConnected :: DType a => Symbol a -> Symbol a -> Symbol a -> Int -> IO (Symbol a)

correlation :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

activation :: DType a => Symbol a -> String -> IO (Symbol a)

leakyReLU :: DType a => Symbol a -> String -> IO (Symbol a)

softmaxActivation :: DType a => Symbol a -> IO (Symbol a)

dropout :: DType a => Symbol a -> Float -> IO (Symbol a)

batchNorm :: DType a => Symbol a -> Symbol a -> Symbol a -> Symbol a -> Symbol a -> IO (Symbol a)

instanceNorm :: DType a => Symbol a -> Symbol a -> Symbol a -> Float -> IO (Symbol a)

l2Normalization :: DType a => Symbol a -> Float -> String -> IO (Symbol a)

convolution :: DType a => Symbol a -> Symbol a -> Symbol a -> String -> Int -> IO (Symbol a)

lrn :: DType a => Symbol a -> Float -> Float -> Float -> Int -> IO (Symbol a)

deconvolution :: DType a => Symbol a -> Symbol a -> Symbol a -> String -> Int -> IO (Symbol a)

pooling :: DType a => Symbol a -> String -> String -> IO (Symbol a)

roiPooling :: DType a => Symbol a -> Symbol a -> String -> Int -> IO (Symbol a)

rnn :: DType a => Symbol a -> Symbol a -> Symbol a -> Symbol a -> Int -> Int -> String -> IO (Symbol a)

embedding :: DType a => Symbol a -> Symbol a -> Int -> Int -> IO (Symbol a)

bilinearSampler :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

gridGenerator :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

upSampling :: DType a => [Symbol a] -> Int -> String -> IO (Symbol a)

spatialTransformer :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

linearRegressionOutput :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

logisticRegressionOutput :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

softmaxOutput :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

maeRegressionOutput :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

svmOutput :: DType a => Symbol a -> Symbol a -> Int -> Float -> Bool -> IO (Symbol a)

softmaxCrossEntropy :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

smoothL1 :: DType a => Symbol a -> Float -> IO (Symbol a)

identityAttachKLSparsereg :: DType a => Symbol a -> IO (Symbol a)

makeLoss :: DType a => Symbol a -> Float -> Float -> String -> IO (Symbol a)

blockGrad :: DType a => Symbol a -> IO (Symbol a)

custom :: DType a => [Symbol a] -> String -> IO (Symbol a)

Tensor Symbol

Methods

dot :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

reshape :: DType a => Symbol a -> [Int] -> IO (Symbol a)

transpose :: DType a => Symbol a -> IO (Symbol a)

(+.) :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

(-.) :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

(*.) :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

(/.) :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

(^.) :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

(.+) :: DType a => Symbol a -> a -> IO (Symbol a)

(.-) :: DType a => Symbol a -> a -> IO (Symbol a)

(.*) :: DType a => Symbol a -> a -> IO (Symbol a)

(./) :: DType a => Symbol a -> a -> IO (Symbol a)

(.^) :: DType a => Symbol a -> a -> IO (Symbol a)

(..-) :: DType a => a -> Symbol a -> IO (Symbol a)

(../) :: DType a => a -> Symbol a -> IO (Symbol a)

(..^) :: DType a => a -> Symbol a -> IO (Symbol a)

(.+=) :: DType a => Symbol a -> a -> IO ()

(.-=) :: DType a => Symbol a -> a -> IO ()

(.*=) :: DType a => Symbol a -> a -> IO ()

(./=) :: DType a => Symbol a -> a -> IO ()

(.^=) :: DType a => Symbol a -> a -> IO ()

_Maximum :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

_Minimum :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

equal :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

notEqual :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

greater :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

greaterEqual :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

lesser :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

lesserEqual :: DType a => Symbol a -> Symbol a -> IO (Symbol a)

_Maximum' :: DType a => Symbol a -> a -> IO (Symbol a)

_Minimum' :: DType a => Symbol a -> a -> IO (Symbol a)

equal' :: DType a => Symbol a -> a -> IO (Symbol a)

notEqual' :: DType a => Symbol a -> a -> IO (Symbol a)

greater' :: DType a => Symbol a -> a -> IO (Symbol a)

greaterEqual' :: DType a => Symbol a -> a -> IO (Symbol a)

lesser' :: DType a => Symbol a -> a -> IO (Symbol a)

lesserEqual' :: DType a => Symbol a -> a -> IO (Symbol a)

DType a => Floating (Symbol a)

Methods

pi :: Symbol a

exp :: Symbol a -> Symbol a

log :: Symbol a -> Symbol a

sqrt :: Symbol a -> Symbol a

(**) :: Symbol a -> Symbol a -> Symbol a

logBase :: Symbol a -> Symbol a -> Symbol a

sin :: Symbol a -> Symbol a

cos :: Symbol a -> Symbol a

tan :: Symbol a -> Symbol a

asin :: Symbol a -> Symbol a

acos :: Symbol a -> Symbol a

atan :: Symbol a -> Symbol a

sinh :: Symbol a -> Symbol a

cosh :: Symbol a -> Symbol a

tanh :: Symbol a -> Symbol a

asinh :: Symbol a -> Symbol a

acosh :: Symbol a -> Symbol a

atanh :: Symbol a -> Symbol a

log1p :: Symbol a -> Symbol a

expm1 :: Symbol a -> Symbol a

log1pexp :: Symbol a -> Symbol a

log1mexp :: Symbol a -> Symbol a

DType a => Fractional (Symbol a)

Methods

(/) :: Symbol a -> Symbol a -> Symbol a

recip :: Symbol a -> Symbol a

fromRational :: Rational -> Symbol a

DType a => Num (Symbol a)

Methods

(+) :: Symbol a -> Symbol a -> Symbol a

(-) :: Symbol a -> Symbol a -> Symbol a

(*) :: Symbol a -> Symbol a -> Symbol a

negate :: Symbol a -> Symbol a

abs :: Symbol a -> Symbol a

signum :: Symbol a -> Symbol a

fromInteger :: Integer -> Symbol a

DType a => Show (Symbol a)

Methods

showsPrec :: Int -> Symbol a -> ShowS

show :: Symbol a -> String

showList :: [Symbol a] -> ShowS

variable

Arguments

:: DType a

=> String

Name.

-> IO (Symbol a)

Result variable.

Make a new symbolic variable with given name.

getName :: DType a => Symbol a -> IO String

Get the name of a given variable.

getAttr :: DType a => Symbol a -> String -> IO (Maybe String)

Get specified attribute of symbol.

setAttr :: DType a => Symbol a -> String -> String -> IO ()

Set specified attribute of symbol.

infershape :: DType a => Symbol a -> [String] -> IO ([[Int]], [[Int]], [[Int]])

Infer the shape of the given symbol, return the in, out and auxiliary shape size.

grad :: DType a => Symbol a -> [String] -> IO (Symbol a)

Get the autodiff of current symbol. This function can only be used if current symbol is a loss function.

bind :: DType a => Symbol a -> Context -> HashMap String (NDArray a) -> IO (Executor a)

Bind with explicit argument mapping (name -- value mapping).

bind' :: DType a => Symbol a -> Context -> [NDArray a] -> IO (Executor a)

Bind without explicit argument mapping (name -- value mapping).

listInputs :: DType a => Symbol a -> IO [String]

List all input arguments.

listOutputs :: DType a => Symbol a -> IO [String]

List all output results.

listAuxiliaries :: DType a => Symbol a -> IO [String]

List all auxiliary states.

Executor

data Executor a

Type alias for variable.

makeExecutor :: DType a => ExecutorHandle -> IO (Executor a)

Make an executor using the given handler.

forward

Arguments

:: DType a

=> Executor a

The executor handle.

-> Bool

Whether this forward is for evaluation purpose.

-> IO ()

Executor forward method.

backward

Arguments

:: DType a

=> Executor a

The executor handle.

-> IO ()

Executor backward method.

getOutputs :: DType a => Executor a -> IO [NDArray a]

DType

module MXNet.Core.Base.DType

Heterogeneous Dictionary.

module MXNet.Core.Base.HMap