Creates a dataset with at most count elements from this dataset
dataset_take(dataset, count)A dataset
Integer representing the number of elements of this dataset that
should be taken to form the new dataset. If count is -1, or if count is
greater than the size of this dataset, the new dataset will contain all
elements of this dataset.
A dataset
Other dataset methods: dataset_batch,
dataset_cache,
dataset_concatenate,
dataset_decode_delim,
dataset_filter,
dataset_interleave,
dataset_map_and_batch,
dataset_map,
dataset_padded_batch,
dataset_prefetch_to_device,
dataset_prefetch,
dataset_repeat,
dataset_shuffle_and_repeat,
dataset_shuffle, dataset_skip