AI 引擎 API 支持对 aie::vector
执行矢量缩减运算。
-
aie::reduce_add
- 返回输入矢量中的元素总和。
-
aie::reduce_add_v
- 返回输入矢量中的元素总和。每个输入矢量的总和存储在输出矢量的一个元素中。
-
aie::reduce_max
- 返回来自含最大值的输入矢量的元素。
-
aie::reduce_min
- 返回来自含最小值的输入矢量的元素。
-
aie::reduce_mul
- 返回输入矢量中的元素的乘法。
矢量缩减示例如下所示。
aie::vector<int16,8> iv,iv2,iv3,iv4;
int16 iv_add=aie::reduce_add(iv);
aie::vector<int16,8> iv_add_v4=aie::reduce_add_v(iv,iv2,iv3,iv4);
int16 iv_max=aie::reduce_max(iv);
int16 iv_min=aie::reduce_min(iv);
int16 iv_mul=aie::reduce_mul(iv);