Used for shrink memories state in DyRNN. The height of state could
be shrinked after running a step block.