A transformation that stops dataset iteration based on a predicate.
dataset_take_while(dataset, predicate, name = NULL)A TF Dataset
A TF dataset
A function that maps a nested structure of tensors (having
shapes and types defined by self$output_shapes and self$output_types)
to a scalar tf.bool tensor.
(Optional.) A name for the tf.data operation.
Example usage:
range_dataset(from = 0, to = 10) %>%
dataset_take_while( ~ .x < 5) %>%
as_array_iterator() %>%
iterate(simplify = FALSE) %>% str()
#> List of 5
#> $ : num 0
#> $ : num 1
#> $ : num 2
#> $ : num 3
#> $ : num 4
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_reduce(),
dataset_repeat(),
dataset_shuffle(),
dataset_shuffle_and_repeat(),
dataset_skip(),
dataset_take(),
dataset_window()