Deep learning和tensorflow学习记录(十): tf.train.string_input_producer

 
 


tf.train.string_input_producer(
    string_tensor,
    num_epochs=None,
    shuffle=True,
    seed=None,
    capacity=32,
    shared_name=None,
    name=None,
    cancel_op=None
)

定义于:tensorflow/python/input.py。

输出字符串到一个输入管道队列。

注意:如果num_epochs不是None,则此函数创建本地计数器 epochs使用local_variables_initializer()初始化局部变量。

参数:

string_tensor:1-D字符串Tensor。

num_epochs:一个整数(可选)。如果指定,string_input_producer在产生OutOfRange错误之前从string_tensor中产生num_epochs次字符串。如果未指定,则可以无限次循环遍历字符串。

shuffle:布尔值。如果为true,则在每个epoch内随机打乱顺序。

seed:一个整数(可选)。如果shuffle==True,则使用种子。

capacity:一个整数。设置队列容量。

shared_name:(可选的)。如果设置,则此队列将在多个会话的给定名称下共享。对具有此队列的设备打开的所有会话都可以通过shared_name访问它。在分布式设置中使用它意味着只有能够访问此操作的其中一个会话才能看到每个名称。

name:此操作的名称(可选)。

cancel_op:取消队列的操作(可选)。

返回:

一个带有输出字符串的队列。此队列的一个QueueRunner被添加到当前Graph的QUEUE_RUNNER集合中。

猜你喜欢

转载自blog.csdn.net/heiheiya/article/details/80967301
今日推荐