Python tf.data.experimental.RandomDataset.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.experimental.RandomDataset.take_while。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。