Python tf.data.TFRecordDataset.take_while用法及代码示例

用法

take_while(
    predicate, name=None
)

predicate 将张量的嵌套结构(具有由 self.output_shapes 和 self.output_types 定义的形状和类型)映射到标量 tf.bool 张量的函数。
name (可选。) tf.data 操作的名称。

基于 predicate 停止数据集迭代的转换。

dataset = tf.data.Dataset.range(10)
dataset = dataset.take_while(lambda x:x < 5)
list(dataset.as_numpy_iterator())
[0, 1, 2, 3, 4]

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.data.TFRecordDataset.take_while。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。