关于onnx转为om模型后,ReduceMean算子的输出问题
收藏回复举报
关于onnx转为om模型后,ReduceMean算子的输出问题
t('forum.solved') 已解决
发表于2026-02-07 13:04:04
0 查看

请教一下各位大佬,我用ATC将onnx转为om模型后,ReduceMean的输出期望是计算-1维均值 得到output shapes[1,1,8,1],但实际是[1,1,1,1] 猜测可能是对8*128个数据一起做了均值处理,想确认下这个算子是不是只支持对全部数据进行均值计算,无法对固定维度实现均值计算

cke_236.png

注:我在onnx中,是定义了他的均值计算维度是3的,也就是-1维

cke_19509.png

我要发帖子