MXNet.Core.Base.Executor

Symbol module.

Documentation

newtype Executor a

Type alias for variable.

Constructors

Executor

Fields

• getHandle :: ExecutorHandle

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]