1. 程式人生 > >【Tensorflow】(十): tf.train.string_input_producer

【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集合中。