PyTorch 2.0正式版发布!一行代码提速2倍,100%向后兼容(1/7)

更多
2023-3-26 16:57 164人参与 0条评论 自动播放 开灯

PyTorch 2.0正式版发布!一行代码提速2倍,100%向后兼容

编译,还是编译! PyTorch 2.0的最新编译器技术包含:TorchDynamo、AOTAutograd、PrimTorch和TorchInductor。一切这些都是用Python开发的,而不是C++(Python与之兼容)。并且还支持dynamic shape,无需重新编译就能发送不同大小的向量,灵活且易学。TorchDynamo它能够借助Python Frame Evaluation Hooks,保险地获取PyTorch程序,这项严重创新是PyTorch过去 5 年来在保险图结构捕获 (saf ... [查看原文]

上传图片 点击发表评论 (164人参与 0条评论)