tf.train.slice_input_producer

代码语言：javascript复制

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

tensorflow

0 人点赞