dataset_reduce: Reduces the input dataset to a single element.

In tfdatasets: Interface to 'TensorFlow' Datasets

Reduces the input dataset to a single element.

Description

The transformation calls reduce_func successively on every element of the input dataset until the dataset is exhausted, aggregating information in its internal state. The initial_state argument is used for the initial state and the final state is returned as the result.

Usage

dataset_reduce(dataset, initial_state, reduce_func)

Arguments

dataset	A dataset
initial_state	An element representing the initial state of the transformation.
reduce_func	A function that maps ⁠(old_state, input_element)⁠ to new_state. It must take two arguments and return a new element. The structure of new_state must match the structure of initial_state.

Value

A dataset element.

See Also

Other dataset methods: dataset_batch(), dataset_cache(), dataset_collect(), dataset_concatenate(), dataset_decode_delim(), dataset_filter(), dataset_interleave(), dataset_map(), dataset_map_and_batch(), dataset_padded_batch(), dataset_prefetch(), dataset_prefetch_to_device(), dataset_repeat(), dataset_shuffle(), dataset_shuffle_and_repeat(), dataset_skip(), dataset_take(), dataset_take_while(), dataset_window()

tfdatasets documentation built on Sept. 11, 2024, 8:53 p.m.