vreducev2
函数功能
vreducev2指令抽象。
vreducev2指令根据输入y向量的mask数据，决定取x向量中的某些block级的元素存至z向量，由于mskpp中的tensor并无相关概念，因此增加了reserve_num的参数，z输出的shape由该参数决定。
函数原型
class vreducev2(x, y, z, reserve_num)
参数说明
参数名 |
输入/输出 |
数据类型 |
说明 |
|---|---|---|---|
x |
输入 |
tensor变量 |
输入x向量tensor。支持UINT16、UINT32。 |
y |
输入 |
tensor变量 |
输入y向量tensor。支持UINT16、UINT32。 |
reserve_num |
输入 |
int |
指定输出元素的个数。 |
z |
输出 |
tensor变量 |
输出z向量tensor。支持UINT16、UINT32。 |
使用示例
ub_x, ub_y = Tensor("UB"), Tensor("UB")
reserve_num = 16
ub_x.load(gm_x)
out = vreducev2(ub_x, ub_y, reserve_num)()
父主题： mskpp对外接口使用说明