View source on GitHub |
Basic wrapper for the ObtainNextOp.
tf.contrib.input_pipeline.obtain_next(
string_list_tensor, counter
)
Args | |
---|---|
string_list_tensor
|
A tensor that is a list of strings |
counter
|
an int64 ref tensor to keep track of which element is returned. |
Returns | |
---|---|
An op that produces the element at counter + 1 in the list, round robin style. |