tff.analytics.data_processing.get_all_elements

Gets all the elements from the input dataset.

The input dataset must yield batched 1-d tensors. This function reads each coordinate of the tensor as an individual element.

dataset A tf.data.Dataset.

A rank-1 Tensor containing the elements of the input dataset.