特性范围
数据类型
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