Python tf.data.IteratorSpec用法及代码示例

tf.data.Iterator 的类型规范。

继承自：TypeSpec

用法

tf.data.IteratorSpec(
    element_spec
)

属性

element_spec tf.TypeSpec 对象的(嵌套)结构，表示迭代器元素的类型规范。
value_type 与此 TypeSpec 兼容的值的 Python 类型。
特别是，所有与此 TypeSpec 兼容的值都必须是此类型的实例。

例如，tf.data.IteratorSpec 可用于定义一个将 tf.data.Iterator 作为输入参数的 tf.function：

@tf.function(input_signature=[tf.data.IteratorSpec(
  tf.TensorSpec(shape=(), dtype=tf.int32, name=None))])
def square(iterator):
  x = iterator.get_next()
  return x * x
dataset = tf.data.Dataset.from_tensors(5)
iterator = iter(dataset)
print(square(iterator))
tf.Tensor(25, shape=(), dtype=int32)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.data.IteratorSpec。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。