Loading Data

Neural network module.

Datasets

data.Dataset Abstract dataset class.
data.ArrayDataset(*args) A dataset that combines multiple dataset-like objects, e.g.
data.RecordFileDataset(filename) A dataset wrapping over a RecordIO (.rec) file.

Sampling examples

data.Sampler Base class for samplers.
data.SequentialSampler(length[, start]) Samples elements from [start, start+length) sequentially.
data.RandomSampler(length) Samples elements from [0, length) randomly without replacement.
data.BatchSampler(sampler, batch_size[, …]) Wraps over another Sampler and return mini-batches of samples.

Loading examples

data.DataLoader(dataset[, batch_size, …]) Loads data from a dataset and returns mini-batches of data.