def _unstack_ta(inp): return tf.TensorArray( dtype=inp.dtype, size=tf.shape(inp)[0], element_shape=inp.get_shape()[1:]).unstack(inp)