grain.experimental.LimitIterDataset

grain.experimental.LimitIterDataset#

class grain.experimental.LimitIterDataset(parent, count)#

Limits the number of elements in the dataset.

Example usage:

` list(LimitIterDataset(MapDataset.range(5).to_iter_dataset(), 2) == [0, 1] `

Parameters:
parent#

The dataset to limit.

count#

The maximum number of elements to include in the dataset.

__init__(parent, count)#

Initializes the limit dataset.

Parameters:

Methods

__init__(parent, count)

Initializes the limit dataset.

batch(batch_size, *[, drop_remainder, batch_fn])

Returns a dataset of elements batched along a new first dimension.

filter(transform)

Returns a dataset containing only the elements that match the filter.

map(transform)

Returns a dataset containing the elements transformed by transform.

map_with_index(transform)

Returns a dataset of the elements transformed by the transform.

mp_prefetch([options, worker_init_fn])

Returns a dataset prefetching elements in multiple processes.

pipe(func, /, *args, **kwargs)

Syntactic sugar for applying a callable to this dataset.

prefetch(multiprocessing_options)

Deprecated, use mp_prefetch instead.

random_map(transform, *[, seed])

Returns a dataset containing the elements transformed by transform.

seed(seed)

Returns a dataset that uses the seed for default seed generation.

Attributes

parents

Contents