Return to PyTorch4