description: A transformation that stops dataset iteration based on a predicate.

tf.data.experimental.take_while

A transformation that stops dataset iteration based on a predicate.

tf.data.experimental.take_while(
    predicate
)

Args

predicate 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.

Returns

A Dataset transformation function, which can be passed to tf.data.Dataset.apply.