跳到主要内容

MindSpore Lite Kit算子支持列表

本文档介绍了MindSpore Lite Kit与ONNX Opset18相比所支持的CPU后端算子列表以及对应关系。在使用模型转换工具将ONNX模型转换为ms模型进行部署时,通过查询此表可以知道MindSpore Lite Kit所支持的常用ONNX算子,以确保模型转换成功。

  • 以下所有算子,均不支持int64类型输入。
MindSpore Lite算子名称算子功能对应ONNX算子名称算子规格
Abs逐元素计算绝对值Abs不支持uint8类型。不支持输入张量量化参数为空。
Activation激活函数Celu Clip Elu Gelu HSigmoid LeakyRelu PRelu Relu Sigmoid SoftMax SoftPlus Tanh-
AddFusion逐元素计算加法Add-
ArgMaxFusion求某一维度最大值ArgMax不支持uint8类型。不支持输入张量量化参数为空。
AvgPoolFusion平均池化AveragePool GlobalAveragePool GlobalMaxPool MaxPool-
BatchNorm批量归一化BatchNormalization-
BroadcastTo扩维Expand-
Cast数据类型转换Cast不支持以下数值类型转换:fp32转int8、fp32转uint32、int32转int8、int32转uint32、int32转uint8、int8转bool、int8转uint8。
Ceil向上取整Ceil-
Clip限制元素范围Clip-
Concat拼接张量Concat-
Conv2DFusion2D卷积Conv-
Cos逐元素计算余弦Cos-
CumSum累计元素和CumSum-
DepthToSpace将深度数据重新排列到空间维度中DepthToSpace不支持uint8类型。不支持未知维度输入。
DivFusion逐元素除法Div不支持除数为0。
Eltwise元素级求和Sum-
Equal判断输入是否相等Equal不支持uint8输入;int8输入不支持bool输出。
Erf误差函数Erf-
ExpFusion逐元素取指数Exp-
Flatten数据按维度展开Flatten不支持uint8类型。
Floor向下取整Floor-
FusedBatchNorm对输入做标准化BatchNormalization-
Gather沿单一维度收集指定索引位置的元素Gather不支持uint8类型。不支持QuantType_QUANT_NONE量化类型。
GatherD将输入tensor中的元素根据索引tensor进行收集GatherElements-
GatherNd将输入张量的切片聚合成具有indices指定维度的新张量GatherND从HarmonyOS 6.1.0开始支持空shape推理。
InstanceNorm实例归一化InstanceNormalization-
Log逐元素求对数Log不支持负数输入。
LogicalNot元素级逻辑非Not-
LogSoftmax对输入向量进行softmax操作,然后再对softmax结果取对数LogSoftmax不支持inf输入。
LRN局部响应标准化,用于防止数据过度拟合LRN-
MatMulFusion对2个输入做矩阵乘法运算;使用输入张量、一组学习的权重计算内积,并添加偏差Gemm MatMul-
Maximum取元素级最大值Max-
MaxPoolFusion最大池化GlobalMaxPool MaxPool-
Minimum取元素级最小值Min-
Mod返回除法元素的余数Mod-
MulFusion逐元素乘法Mul-
Neg逐元素求负数Neg-
PadFusion将输入张量加上指定的 padding,使其达到指定的大小Pad不支持int32类型。
PowFusion逐元素求幂Pow仅支持指数为单个常数。
PReLUFusionPRelu激活函数PRelu-
Range生成某个区间内的元素Range-
Reciprocal返回倒数Reciprocal-
Reduce对指定维度进行聚合计算来减少张量维度ReduceSum ReduceMean ReduceMax ReduceMin ReduceProd ReduceLogSum ReduceLogSumExp ReduceSumSquare ReduceL1 ReduceL2从HarmonyOS 6.1.0开始支持空shape推理。
Reshape改变张量形状,总元素个数不变Reshape-
Round四舍五入到最接近的整数数值Round-
ScatterNdUpdate使用给定值以及输入索引更新输入数据的值ScatterND-
Shape获得张量shapeShape-
Sin逐元素计算正弦Sin-
Size获取张量维度大小Size-
SliceFusion张量切片操作Slice-
Softmax归一化操作Softmax-
SpaceToDepth高度和宽度维度的值移至深度维度SpaceToDepth-
Sqrt逐元素开根号Sqrt-
Squeeze移除维度为1的维度Squeeze-
StridedSliceTensor切片Slice-
SubFusion逐元素相减Sub-
TileFusion平铺给定矩阵Tile不支持int8类型。
TopKFusion从输入张量中返回top K个元素TopK-
TransposeTensor转置Transpose-
Tril下三角矩阵Trilu(属性upper=0)-
Triu上三角矩阵Trilu(属性upper=1)-
Unsqueeze将输入张量添加一个新的维度Unsqueeze-
Where元素选择Where-