當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

Python tf.compat.v1.data.experimental.RandomDataset.take_while用法及代碼示例

用法

take_while(
    predicate, name=None
)

參數

  • predicate 將張量的嵌套結構(具有由 self.output_shapes 和 self.output_types 定義的形狀和類型)映射到標量 tf.bool 張量的函數。
  • name (可選。) tf.data 操作的名稱。

返回

  • 一個Dataset。

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