mxnet-0.2.0.0: MXNet interface in Haskell.

Copyright(c) 2016-2017 Tao He
LicenseMIT
Maintainersighingnow@gmail.com
Safe HaskellNone
LanguageHaskell2010

MXNet.Core.Base.NDArray

Description

NDArray module, provide an imperative-style programming interface.

Synopsis

Documentation

newtype NDArray a Source #

NDArray type alias.

Constructors

NDArray 

Instances

Neural NDArray Source # 

Methods

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Tensor NDArray Source # 

Methods

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

(DType a, (~) * a Int8) => Eq (NDArray Int8) Source # 
(DType a, (~) * a Int32) => Eq (NDArray Int32) Source # 
(DType a, Floating a) => Eq (NDArray a) Source # 

Methods

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

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

DType a => Floating (NDArray a) Source # 

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) Source # 

Methods

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

recip :: NDArray a -> NDArray a #

fromRational :: Rational -> NDArray a #

DType a => Num (NDArray a) Source # 

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) Source # 

Methods

showsPrec :: Int -> NDArray a -> ShowS #

show :: NDArray a -> String #

showList :: [NDArray a] -> ShowS #

waitAll :: IO () Source #

Wait all async operation to finish in MXNet.

makeEmptyNDArray Source #

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 Source #

Arguments

:: DType a 
=> [Int]

size of every dimensions.

-> Context 
-> Vector a 
-> IO (NDArray a) 

Make a new NDArray with given shape.

ndshape Source #

Arguments

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

Dimensions and size of every dimensions.

Get the shape of given NDArray.

ndsize Source #

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 Source #

Get context of the given ndarray.

copy :: DType a => NDArray a -> IO (NDArray a) Source #

Make a copy of the give ndarray.

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

Get data stored in NDArray.

slice Source #

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.

at Source #

Arguments

:: DType a 
=> NDArray a 
-> Int

The index.

-> NDArray a 

Return a sub ndarray that shares memory with current one.

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

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

onehotEncode Source #

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 Source #

Arguments

:: DType a 
=> [Int]

Shape.

-> IO (NDArray a) 

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

ones Source #

Arguments

:: DType a 
=> [Int]

Shape.

-> IO (NDArray a) 

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

full Source #

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 Source #

Arguments

:: DType a 
=> [Int]

Shape.

-> Vector a 
-> IO (NDArray a) 

Create a new NDArray that copies content from source_array.

data PrettyWrapper Source #

Wrapper for pretty print multiple dimensions matrices.

Constructors

Pretty a => MkPretty 

Fields