特性范围
数据类型
src数据类型 |
dst数据类型 |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
s8 |
s8 |
u8 |
u8 |
mean（均值），variance（方差），Scale（放缩），Shift（偏移）的数据类型与src和dst无关，固定为f32。
数据排布
支持3D~5D Tensor维度，具体数据排布如下，输入和输出排布相同。
Tensor维度 |
src数据排布 |
dst数据排布 |
|---|---|---|
2D Tensor |
ab |
ab |
3D Tensor |
abc、acb |
abc、acb |
4D Tensor |
abcd、acdb |
abcd、acdb |
5D Tensor |
abcde、acdeb |
abcde、acdeb |
父主题： Group Normalization