tf.raw_ops.TakeWhileDataset | TensorFlow v2.16.1 (original) (raw)

tf.raw_ops.TakeWhileDataset

Stay organized with collections Save and categorize content based on your preferences.

Creates a dataset that stops iteration when predicate` is false.

View aliases

Compat aliases for migration

tf.compat.v1.raw_ops.TakeWhileDataset

tf.raw_ops.TakeWhileDataset(
    input_dataset,
    other_arguments,
    predicate,
    output_types,
    output_shapes,
    metadata='',
    name=None
)

The predicate function must return a scalar boolean and accept the following arguments:

One tensor for each component of an element of input_dataset.
One tensor for each value in other_arguments.

Args
input_dataset	A Tensor of type variant.
other_arguments	A list of Tensor objects. A list of tensors, typically values that were captured when building a closure for predicate.
predicate	A function decorated with @Defun. A function returning a scalar boolean.
output_types	A list of tf.DTypes that has length >= 1.
output_shapes	A list of shapes (each a tf.TensorShape or list of ints) that has length >= 1.
metadata	An optional string. Defaults to "".
name	A name for the operation (optional).

Returns
A Tensor of type variant.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.