code: def state_size(self):
      
      return LSTMState(tf.TensorShape([self._hidden_state_size]),
                       tf.TensorShape([self._hidden_size]))
docs: Tuple of `tf.TensorShape`s indicating the size of state tensors.