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

算子范围

KAIL_DNN目前支持的算子如表1所示。

表1 KAIL_DNN支持的算子

算子名称

算子说明

Eltwise

逐元素操作算子。

Layer Normalization

层归一化算子。

Inner Product

矩阵内积算子。

Softmax

Softmax归一化算子。

Sum

求和算子。

Matmul

矩阵乘法算子。

Convolution

卷积算子。

Deconvolution

计算反卷积操作。

Concat

拼接算子。

Resampling

数据重新采样放缩算子。

Shuffle

数据混洗算子。

Reorder

数据重新排布算子。

Pool

池化操作算子。

Batch Normalization(bnormal)

批次归一化算子。

Local Response Normalization(lrn)

局部响应归一化算子。

Reduction

推断算子。

PReLU

使用训练alpha的激活算子(Leaky ReLU)。

Binary

二进制基元计算张量算子。

RNN

循环神经网络算子。