AI开发平台ModelArts-编写自定义算子

时间:2023-11-01 16:20:20

编写自定义算子

用户通过自定义算子功能,可以实现个性化的算子编写。

用户单击“新增自定义算子”图标,新建并打开一个模板算子,即一个算子编辑器(相当于Ipython Notebook的一个cell),输入自定义算子名称,即可以在新建的算子编辑器里面实现自定义算子开发,如图1所示。用户可进行算子编辑、调试、复制、保存等功能。

图1 新增自定义算子

由于算子使用过程中会将ipynb文件中的所有Cell合并为一个Cell,因此建议开发者直接将算子代码写在一个Cell中。

support.huaweicloud.com/devtool-modelarts/devtool-modelarts_0033.html