TensorFlow 2 version
|
View source on GitHub
|
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.
|