Hi @FAlex,
thanks for pointing out the potential for improvement in the documentation!
I’ve put this into a PR on github 62, so hopefully PyTorch 1.0 ships with clearer documentation.
Best regards
Thomas