vector_dup
函数功能
vector_dup指令抽象。
y = fill(x)， x、 y按元素填充元素。
函数原型
class VectorDup(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作为模拟标量输入，不增加性能开销。
使用示例
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对外接口使用说明