code: def state_size(self):
      
      hidden_size = tf.TensorShape(
          self._input_shape[:-1] + (self._output_channels,))
      return (hidden_size, hidden_size)
docs: Tuple of `tf.TensorShape`s indicating the size of state tensors.