def create_ta(s): return tf.TensorArray(dtype=s.dtype, size=self._train_interval + 1, element_shape=tf.TensorShape( [batch_size]).concatenate(s.shape))
def create_ta(s): return tf.TensorArray(dtype=s.dtype, size=maximum_iterations, element_shape=tf.TensorShape( [batch_size]).concatenate(s.shape))