Fixed incorrect description of what each dimension represents

[vaharoni]

Fixes #3622

Description

Suggestion for tweaking the text containing a possible minor inaccuracy.

The basic Build the Neural Network tutorial has the text: Calling the model on the input returns a 2-dimensional tensor with dim=0 corresponding to each output of 10 raw predicted values for each class, and dim=1 corresponding to the individual values of each output.

I'm suggesting the following: Calling the model on the input returns a 2-dimensional tensor with dim=0 corresponding to the batch dimension (one entry per example in the input) and dim=1 corresponding to the 10 raw predicted class scores (logits) for each example.

Checklist

• [x] The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
• [x] Only one issue is addressed in this pull request
• [x] Labels from the issue that this PR is fixing are added to this pull request
• [x] No unnecessary issues are included into this pull request.