Safe Haskell	None
Language	Haskell2010

Torch.Data.Dataset

Synopsis

data CollatedDataset (m :: Type -> Type) dataset batch collatedBatch = CollatedDataset {
- set :: dataset
- chunkSize :: Int
- collateFn :: Pipe [batch] collatedBatch m ()
}

Documentation

data CollatedDataset (m :: Type -> Type) dataset batch collatedBatch Source #

This type is actually not very useful. | It would actually be better to define a transform | on top of another dataset, since then we can do this in parallel

Constructors

CollatedDataset

Fields

set :: dataset
chunkSize :: Int
collateFn :: Pipe [batch] collatedBatch m ()

Instances

Instances details

Datastream m seed dataset batch => Datastream m seed (CollatedDataset m dataset batch collatedBatch) collatedBatch Source #

Instance details

Defined in Torch.Data.Dataset

Methods

streamSamples :: CollatedDataset m dataset batch collatedBatch -> seed -> ListT m collatedBatch Source #