鲲鹏社区首页
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

算子说明

KAIL_DNN_EXT是鲲鹏深度学习性能加速库KAIL_DNN的拓展模块,有以下特点:

  • 接口易用:使用Cython框架,向用户提供Python语言接口,更容易嵌入用户场景。
  • 高性能:底层使用C代码实现,接口性能好。

当前支持以下算子:

  • random_choice算子
  • softmax算子