MXNet.Core.NDArray

NDArray module.

Data type definitions

type NDArray = NDArrayHandle

NDArray type alias.

data Context

Context definition.

• DeviceType

  1. cpu
  2. gpu
  3. cpu_pinned

Instances

Eq Context

Methods

(==) :: Context -> Context -> Bool

(/=) :: Context -> Context -> Bool

Show Context

Methods

showsPrec :: Int -> Context -> ShowS

show :: Context -> String

showList :: [Context] -> ShowS

Functions about NDArray

makeNDArray

Arguments

:: [Int]

size of every dimensions.

-> IO NDArray

Make a new NDArray with given shape.

getNDArrayShape

Arguments

:: NDArray

-> IO (Int, [Int])

Dimensions and size of every dimensions.

Get the shape of given NDArray.

Default contexts

defaultContext :: Context

Default context, use the CPU 0 as device.

contextCPU :: Context

Context for CPU 0.

contextGPU :: Context

Context for GPU 0.