TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.data.experimental.take_while

TensorFlow 1 version View source on GitHub

A transformation that stops dataset iteration based on a predicate.

Aliases:

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.