ReduceMeanV2

产品支持情况

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√

功能说明

• 算子功能：对输入张量x沿指定维度计算给定张量的求平均。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
self	输入	x	float、half	ND
axis	属性	axes	int	ND
keepdims	属性	keepdims	int	ND
y	输出	y	float、half	ND

约束说明

无

待更新

目前支持float32、float16数据类型，仅支持2维，axis仅支持1根轴reduce，keepdims支持0/1，后续会持续改进

调用说明

调用方式	样例代码	说明
aclnn接口	test_reduce_mean_v2	通过[aclnnReduceMeanV2]接口方式调用reduce_mean算子。