Return to PyTorch3