华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
请教一下各位大佬,我用ATC将onnx转为om模型后,ReduceMean的输出期望是计算-1维均值 得到output shapes[1,1,8,1],但实际是[1,1,1,1] 猜测可能是对8*128个数据一起做了均值处理,想确认下这个算子是不是只支持对全部数据进行均值计算,无法对固定维度实现均值计算
注:我在onnx中,是定义了他的均值计算维度是3的,也就是-1维
我要发帖子
请教一下各位大佬,我用ATC将onnx转为om模型后,ReduceMean的输出期望是计算-1维均值 得到output shapes[1,1,8,1],但实际是[1,1,1,1] 猜测可能是对8*128个数据一起做了均值处理,想确认下这个算子是不是只支持对全部数据进行均值计算,无法对固定维度实现均值计算
注:我在onnx中,是定义了他的均值计算维度是3的,也就是-1维