need help with code refactoring in pytorch

there is a code that using pytorch lstm, need to refacor it to use lstmcell & validate it+explanation