Tensor Indexing
Use tensor indexing to retrieve the model's final word prediction.
We'll cover the following
Chapter Goals:
- Extract the word predictions for the final time step of each sequence
Get hands-on with 1400+ tech skills courses.