[BUG] - Minor: incorrect description in the Build the Neural Network tutorial
Open
Nobody has claimed this yet.
bug
- Dominant language
- Python
- Stars
- 9.3k
- Forks
- 4.4k
- Avg merge
- 1d 21h
- Merged PRs (30d)
- 4
Description
Add Link
https://docs.pytorch.org/tutorials/beginner/basics/buildmodel_tutorial.html
Describe the bug
The tutorial has the following paragraph:
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 believe this is inaccurate. dim=0 should correspond to the batch dimension, and dim=1 should have a size of 10, corresponding to the logits of each class.
I'll be submitting a PR shortly with suggested copy, assuming you agree with my observation.
Describe your environment
N/A
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Open the linked Build the Neural Network tutorial and locate the quoted paragraph about the returned two-dimensional tensor. Check its description against the stated batch and class-logit dimensions. Done means correcting that paragraph so dim=0 and dim=1 are described accurately.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- documentation, machine-learning
- Issue type
- Documentation
- Difficulty
- 1/5
- Estimated time
- Under an hour
- Activity status
- Stale
- Clarity
- Clearly specified
- Newbie friendliness
- 55/100