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

Python tf.data.TFRecordDataset.range用法及代碼示例


用法

@staticmethod
range(
    *args, **kwargs
)

參數

  • *args 遵循與 python 的範圍相同的語義。 len(args) == 1 -> start = 0，stop = args[0]，step = 1。len(args) == 2 -> start = args[0]，stop = args[1]，step = 1 . len(args) == 3 -> 開始 = args[0]，停止 = args[1]，步驟 = args[2]。
  • **kwargs
    • output_type：它的預期數據類型。 (可選，默認值：tf.int64)。
    • name：(可選。) tf.data 操作的名稱。

返回

  • Dataset 一個RangeDataset。

拋出

  • ValueError 如果 len(args) == 0。

創建 step-separated 值範圍的 Dataset。

list(Dataset.range(5).as_numpy_iterator())
[0, 1, 2, 3, 4]
list(Dataset.range(2, 5).as_numpy_iterator())
[2, 3, 4]
list(Dataset.range(1, 5, 2).as_numpy_iterator())
[1, 3]
list(Dataset.range(1, 5, -2).as_numpy_iterator())
[]
list(Dataset.range(5, 1).as_numpy_iterator())
[]
list(Dataset.range(5, 1, -2).as_numpy_iterator())
[5, 3]
list(Dataset.range(2, 5, output_type=tf.int32).as_numpy_iterator())
[2, 3, 4]
list(Dataset.range(1, 5, 2, output_type=tf.float32).as_numpy_iterator())
[1.0, 3.0]

相關用法


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