当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.data.Dataset.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.Dataset.range。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。