实现将输入src中的内容拷贝到输入dst指定位置中的功能。
例如:dst: (a,b,c), src: (d,e,f)
dst[starts[0]: ends[0], starts[1]: ends[1], starts[2]: ends[2]] = src
其中,starts[0]: ends[0]长度为src第0维的维度大小,starts[1]: ends[1]长度为src第1维的维度大小,starts[2]: ends[2]长度为src第2维的维度大小。
struct SetValueParam { SVector<int64_t> starts; SVector<int64_t> ends; SVector<int64_t> strides; };
成员名称 |
描述 |
---|---|
starts |
每个轴的start。 |
ends |
每个轴的end。 |
strides |
拷贝时步长,当前仅支持strides为全1。 |
参数 |
维度 |
数据类型 |
格式 |
---|---|---|---|
src |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/float/int32/int64 |
ND |
dst |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/float/int32/int64 |
ND |
参数 |
维度 |
数据类型 |
格式 |
---|---|---|---|
dst |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/float/int32/int64 |
ND |
输出dst与同名输入指向同一地址,即进行原地修改。