MXNet.Core.Base.NDArray

NDArray module, provide an imperative-style programming interface.

Documentation

newtype NDArray a

NDArray type alias.

Constructors

NDArray

Fields

• getHandle :: NDArrayHandle

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.

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

Make a copy of the give ndarray.

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.

at

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 ()

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.

data PrettyWrapper

Wrapper for pretty print multiple dimensions matrices.

Constructors

Pretty a => MkPretty

Fields

• runPretty :: a

Instances

Pretty PrettyWrapper

Destruct pretty

Methods

pPrintPrec :: PrettyLevel -> Rational -> PrettyWrapper -> Doc ann

pPrint :: PrettyWrapper -> Doc ann

pPrintList :: PrettyLevel -> [PrettyWrapper] -> Doc ann