Return to PyTorch2a