昇腾算子库包含了丰富的高性能算子,让神经网络的运行性能更高,昇腾算子库中的算子都是预先实现和编译的,是华为工程师使用昇腾AI处理器架构专用编程语言开发的高度优化的内核函数,能够较好的适配底层硬件架构,具有较高的性能。
一般场景下,开发者无需自己开发算子,但若遇到以下场景,开发者需要考虑进行自定义算子的开发。
例如,针对一个分类应用,我们想从分类模型的推理结果中查找可能性最大的前5个标识,则可以实现一个查找最大值的算子(例如ArgMax),后续就可以直接通过AscendCL接口调用此算子实现对推理结果的后处理。