Gets all the elements from the input dataset.
@tf.function
tff.analytics.data_processing.get_all_elements(
dataset: tf.data.Dataset, string_max_bytes: Optional[int] = None
)
The input dataset
must yield batched rank-1 tensors. This function reads
each coordinate of the tensor as an individual element.
Args |
dataset
|
A tf.data.Dataset .
|
string_max_bytes
|
The maximum length (in bytes) of strings in the dataset.
Strings longer than string_max_bytes will be truncated. Defaults to
None , which means there is no limit of the string length.
|
Returns |
A rank-1 Tensor containing the elements of the input dataset.
|
Raises |
ValueError
|
-- If the shape of elements in dataset is not rank 1.
-- If string_max_bytes is not None and is less than 1.
|
TypeError
|
If dataset.element_spec.dtype must be tf.string is not
tf.string .
|