Return to PyTorch1