昇腾社区首页
EN
注册

PyTorch框架

Pytorch框架单算子模式下的适配插件开发流程和具体样例请参见Ascend Extension for PyTorch 套件与三方库支持清单中的“昇腾自研插件 > 自定义算子适配开发> 基于OpPlugin算子适配开发章节。图模式下自定义算子入图的开发指导请参见PyTorch图模式使用指南(TorchAir)中的“自定义算子入图”章节。