自定义算子

本部分将指导您如何在飞桨中新增算子(Operator,简称Op),也包括一些必要的注意事项。此处的算子是一个广义的概念,包括以下几类:

  1. 原生算子:遵循飞桨框架内部算子开发规范,源码合入到飞桨框架中,与框架一起编译后使用的算子

  2. 外部算子:编写方法较为简洁,不涉及框架内部概念,无需重新编译飞桨框架,以外接模块的方式使用的算子

  3. Python算子:使用Python编写实现前向(forward)和反向(backward)方法,在模型组网中使用的自定义API