vmrgsort4

函数功能

vmrgsort4指令抽象。

函数原型

class Vmrgsort(x, y, z)

参数说明

参数名

输入/输出

数据类型

说明

x

输入

tensor变量

输入x向量tensor,支持FP16、FP32。

y

输入

tensor变量

输入y向量tensor,支持UINT64。

z

输出

tensor变量

输出向量tensor,支持FP16、FP32。

使用示例

ub_x, ub_y, ub_z = Tensor("UB"), Tensor("UB")
ub_x.load(gm_x)
ub_y.load(gm_y)
out = vmrgsort4(ub_x, ub_y, ub_z)()