@property<EOL><INDENT>def state_dim(self) -> int:<DEDENT>
if self.rnn_type == '<STR_LIT>':<EOL><INDENT>return <NUM_LIT:2> * self.hidden_size<EOL><DEDENT>else:<EOL><INDENT>return self.hidden_size<EOL><DEDENT>
Dimension of model state