PyTorch集成