PyTorch框架方式