You should be looking at the docs for the version of pytorch that you have installed, not the docs for the development version.
If you have pytorch 0.3.1, then the docs are at http://pytorch.org/docs/0.3.1/ 101
What shape is the input data?