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

Python tf.compat.v1.data.experimental.SqlDataset.take用法及代碼示例


用法

take(
    count, name=None
)

參數

  • count 一個 tf.int64 標量 tf.Tensor ，表示該數據集的元素數，應用於生成新數據集。如果count 為-1，或者count 大於此數據集的大小，則新數據集將包含此數據集的所有元素。
  • name (可選。) tf.data 操作的名稱。

返回

  • Dataset 一個Dataset。

從該數據集中創建一個最多包含 count 元素的 Dataset。

dataset = tf.data.Dataset.range(10)
dataset = dataset.take(3)
list(dataset.as_numpy_iterator())
[0, 1, 2]

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.compat.v1.data.experimental.SqlDataset.take。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。