特性范围
数据类型
src数据类型 |
dst数据类型 |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
数据排布
支持1D~5D Tensor，支持以下数据排布。
Tensor维度 |
src数据排布 |
dst数据排布 |
|---|---|---|
1D Tensor |
a |
a |
2D Tensor |
ab |
ab |
3D Tensor |
abc |
abc |
4D Tensor |
abcd |
abcd |
5D Tensor |
abcde |
abcde |
参数约束
dst Tensor所被规约的维度值需为1。例如：
- 对于5D src Tensor 5x6x7x8x9与5D dst Tensor 1x1x1x1x1，表示所规约的维度为整个Tensor（第1维-第5维）。
- 对于5D src Tensor 5x6x7x8x9与5D dst Tensor 5x6x7x8x1，表示所规约的维度为最内维（第5维）。
父主题： Reduction