Tensor Indexing

Use tensor indexing to retrieve the model's final word prediction.

Chapter Goals:

  • Extract the word predictions for the final time step of each sequence

Get hands-on with 1400+ tech skills courses.