tf.contrib.training.NextQueuedSequenceBatch.state()

tf.contrib.training.NextQueuedSequenceBatch.state(state_name)

Returns batched state tensors.

Args:

• state_name: string, matches a key provided in initial_states.

Returns:

A Tensor: a batched set of states, either initial states (if this is the first run of the given example), or a value as stored during a previous iteration via save_state control flow. Its type is the same as initial_states["state_name"].dtype. If we had at input:

initial_states[state_name].get_shape() == [d1, d2, ...],

then

state(state_name).get_shape() == [batch_size, d1, d2, ...]

Raises:

• KeyError: if state_name does not match any of the initial states declared in initial_states.