self.decoder(output[:, :, -1]).squeeze() should be self.decoder(output[:, -1, :]).squeeze()

[bjuthjliu]

def forward(self, inputs): # Avoid breaking if the last batch has a different size batch_size = inputs.size(0) if batch_size != self.batch_size: self.batch_size = batch_size

    encoded = self.encoder(inputs)
    output, hidden = self.rnn(encoded, self.init_hidden())
    output = self.decoder(output[:, :, -1]).squeeze()
    return output