tf.train.slice_input_producer

tf.train.slice_input_producer(
    tensor_list,
    num_epochs=None,
    shuffle=True,
    seed=None,
    capacity=32,
    shared_name=None,
    name=None
)

在tensor_list中生成每个张量的切片。使用队列实现——队列的QueueRunner被添加到当前图的QUEUE_RUNNER集合中。

参数:

  • tensor_list: 张量对象列表。tensor_list中的每个张量在第一维中必须具有相同的大小。
  • num_epochs: 一个整数(可选)。如果指定,slice_input_producer将在生成OutOfRange错误之前生成每个片num_epochs次。如果没有指定,slice_input_producer可以无限次循环遍历片。
  • suffle: 布尔。如果为真,则在每个历元内随机打乱整数。
  • seed: 一个整数(可选)。种子用于if shuffle == True。
  • capacity: 一个整数。设置队列容量。
  • shared_name: (可选)。如果设置了,此队列将在多个会话中以给定的名称共享。
  • name: 操作的名称(可选)。

返回值:

  • 张量列表,每个张量对应一个tensor_list元素。如果张量在tensor_list中有形状[N, a, b, ..],则对应的输出张量的形状为[a, b,…,z]。

可能产生的异常:

  • ValueError: if slice_input_producer produces nothing from tensor_list.

原链接: https://tensorflow.google.cn/versions/r1.8/api_docs/python/tf/train/slice_input_producer

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券