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

Python tf.data.experimental.Counter用法及代碼示例


創建一個 Dataset，它從 start 開始計數，步長為 step 。

用法

tf.data.experimental.Counter(
    start=0, step=1, dtype=tf.dtypes.int64
)

參數

  • start (可選。)計數器的起始值。默認為 0。
  • step (可選。)計數器的步長。默認為 1。
  • dtype (可選。)計數器元素的數據類型。默認為 tf.int64 。

返回

  • 標量 dtype 元素的 Dataset。

不像tf.data.Dataset.range 會在某個結束數字處停止，Counter 將無限期地產生元素。

dataset = tf.data.experimental.Counter().take(5)
list(dataset.as_numpy_iterator())
[0, 1, 2, 3, 4]
dataset.element_spec
TensorSpec(shape=(), dtype=tf.int64, name=None)
dataset = tf.data.experimental.Counter(dtype=tf.int32)
dataset.element_spec
TensorSpec(shape=(), dtype=tf.int32, name=None)
dataset = tf.data.experimental.Counter(start=2).take(5)
list(dataset.as_numpy_iterator())
[2, 3, 4, 5, 6]
dataset = tf.data.experimental.Counter(start=2, step=5).take(5)
list(dataset.as_numpy_iterator())
[2, 7, 12, 17, 22]
dataset = tf.data.experimental.Counter(start=10, step=-1).take(5)
list(dataset.as_numpy_iterator())
[10, 9, 8, 7, 6]

相關用法


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