vector_dup

功能说明

vector_dup指令抽象。

y = vector_dup(x)， x、 y按元素填充元素。

接口原型

class vector_dup(x, y, fill_shape)

参数说明

参数名	输入/输出	数据类型	说明
x	输入	tensor变量	输入x向量tensor。支持FP16、FP32、INT16、INT32、UINT16、UINT32。
y	输出	tensor变量	输出y向量tensor。支持FP16、FP32、INT16、INT32、UINT16、UINT32。
fill_shape	输入	list	表示目标tensor的要被扩充的shape值。

约束说明

由于该指令输入仅一个标量，因此需要创建一个shape为[1]且is_inited=True的tensor作为模拟标量输入，不增加性能开销。

使用示例

from mskpp import vector_dup, Tensor
ub_x = Tensor("UB", "FP16", [1], format="ND", is_inited=True)
ub_y = Tensor("UB")
out = vector_dup(ub_x, ub_y, [8, 2048])()

父主题： msKPP对外接口使用说明