MXNet.NN.DataIter.Class

Documentation

type family DatasetConstraint (d :: * -> *) (m :: * -> *) :: Constraint

Constraints on Dataset and the monad where the operation shall be ran.

Instances

type DatasetConstraint LVec m

type DatasetConstraint LVec m = MonadIO m

class Dataset (d :: * -> *) where

Abstract Dataset type class

Minimal complete definition

fromListD, zipD, sizeD, forEachD

Methods

fromListD :: [e] -> d e

Create Dataset from `[]`. note that depending on the instance, it may or may not work with infinitive list.

zipD :: d e1 -> d e2 -> d (e1, e2)

Zip two Datasets

sizeD :: (DatasetConstraint d m, Monad m) => d e -> m Int

Get number of elements

forEachD :: (DatasetConstraint d m, Monad m) => d e -> (e -> m a) -> m [a]

Apply a function on each element of Dataset

forEachD_i :: (DatasetConstraint d m, Monad m) => d e -> ((Int, e) -> m a) -> m [a]

Apply a function on each element of Dataset together with the element's index. Note that the default implmentation assumes the Dataset can be created from a infinitive list.

forEachD_ni :: (DatasetConstraint d m, Monad m) => d e -> (((Int, Int), e) -> m a) -> m [a]

Apply a function on each element of Dataset together with the total number of elements and the element's index.

Instances

Dataset LVec

Methods

fromListD :: [e] -> LVec e

zipD :: LVec e1 -> LVec e2 -> LVec (e1, e2)

sizeD :: (DatasetConstraint LVec m, Monad m) => LVec e -> m Int

forEachD :: (DatasetConstraint LVec m, Monad m) => LVec e -> (e -> m a) -> m [a]

forEachD_i :: (DatasetConstraint LVec m, Monad m) => LVec e -> ((Int, e) -> m a) -> m [a]

forEachD_ni :: (DatasetConstraint LVec m, Monad m) => LVec e -> (((Int, Int), e) -> m a) -> m [a]