PyTorch优化

自动分析、修改并验证 contract.md 中定义的 PyTorch 任务，通过循环迭代实现性能优化，最终输出一个整洁的优化分支。