ReduceMean
功能
计算输入Tensor的指定维度的元素的均值。
如果keepdim等于1,得到的Tensor的秩与输入的相同。如果keepdim等于0,那么得到的Tensor就会被精简维数。
输入
x:输入Tensor,数据类型支持float16、float。
属性
- axes:数据类型为int列表,指定精减的维度。
- keepdims:数据类型为int,默认值为1,是否保留缩减后的维度。
输出
y:输出Tensor,和输入x同样的数据类型和format。
约束与限制
无。
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16。
父主题: 支持ONNX算子清单