ReduceMean

功能

计算输入Tensor的指定维度的元素的均值。

如果keepdims等于1，得到的Tensor的秩与输入的相同。如果keepdims等于0，那么得到的Tensor就会被精简维数。

输入

x：输入Tensor，数据类型支持float16、float。

属性

• axes：数据类型为int列表，指定精减的维度。
• keepdims：数据类型为int，默认值为1，是否保留缩减后的维度。

输出

y：输出Tensor，和输入x同样的数据类型。

约束与限制

axes为空时，当前输出不做全维度规约。此时建议修改onnx算子的axes为所有轴；或者在torch导出onnx图前，在torch模型中使用amax来对所有轴规约（例如，x.amax(dim=[0, 1, 2])）。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16。