PyTorch: Difference between revisions