Build A Large Language Model -from Scratch- Pdf -2021 |work| -

out, _ = self.rnn(self.embedding(x), (h0, c0)) out = self.fc(out[:, -1, :]) return out