Feature Scope
Data Types
src Data Type |
dst Data Type |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
Data Layout
1D to 5D tensors are supported. The following data layout formats are supported.
Tensor Dimension |
src Data Layout |
dst Data Layout |
|---|---|---|
1D |
a |
a |
2D |
ab |
ab |
3D |
abc |
abc |
4D |
abcd |
abcd |
5D |
abcde |
abcde |
Parameter Constraints
The dimension of the dst tensor being reduced must be 1. Examples:
- A 5D src tensor (5×6×7×8×9) paired with a 5D dst tensor (1×1×1×1×1) indicates reduction across all dimensions (dimensions 1 through 5).
- A 5D src tensor (5×6×7×8×9) paired with a 5D dst tensor (5×6×7×8×1) indicates reduction only along the innermost dimension (dimension 5).
Parent topic: Reduction