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