为训练和干扰添加对 pytorch 的支持

请将支持添加到 pytorch 官方分支,因为大家都在学习 pytorch,它比原生版本更容易,并使其支持跨 GPU,例如 AMD 和 Intel Arc。