GroupNormV2

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	√

功能说明

• 算子功能：计算输入x的组归一化结果out，均值meanOut，以及标准差的倒数rstdOut。

• 计算公式：

  • GroupNorm： 记 $E[x]=\bar{x}$代表$x$的均值，$Var[x]=\frac{1}{n}\ast {\sum }_{i=1}^n(x_i-E[x]{)}^2$代表$x$的方差，则

  $$\{\begin{array}{rcl}y& & =\frac{x-E[x]}{\sqrt{Var[x]+eps}}\ast \gamma +\beta \\ mean& & =E[x]\\ rstd& & =\frac{1}{\sqrt{Var[x]+eps}}\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	out计算公式中的x。	FLOAT16、FLOAT、BFLOAT16	ND
gamma	输入	out计算公式中的gamma。	FLOAT16、FLOAT、BFLOAT16	ND
beta	输入	out计算公式中的beta。	FLOAT16、FLOAT、BFLOAT16	ND
num_groups	属性	x的第1维度分为group组。	INT32、INT64	-
data_format	可选属性	输入数据格式，支持"NHWC"。	STRING	-
eps	可选属性	out和rstdOut计算公式中的eps值，默认值为0.00001f。	DOUBLE	-
is_training	可选属性	标记是否训练场景，true表示训练场景，false表示推理场景，默认值为true。	BOOL	-
y	输出	输出张量。	FLOAT16、FLOAT、BFLOAT16	ND
mean	输出	mean输出张量。	FLOAT16、FLOAT、BFLOAT16	ND
rstd	输出	rstd输出张量。	FLOAT16、FLOAT、BFLOAT16	ND

约束说明

无

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_group_norm_v2	通过算子IR构图方式调用GroupNormV2算子。