根据参数与InTensor，生成一个数据重新排布过的OutTensor。
输出y基于输入x的总偏移量要求小于输入x的大小。
struct AsStridedParam {
SVector<int64_t> size;
SVector<int64_t> stride;
SVector<int64_t> offset;
};
成员名称 |
描述 |
|---|---|
size |
OutTensor的shape。 size的长度要求小于或等于8且各元素要求大于0。 |
stride |
用于从InTensor推导OutTensor的各维度的步长。 stride的长度要求与size一致，各元素要求大于或等于0。 |
offset |
OutTensor内存相对于InTensor内存的偏移，作为常数使用。 offset的长度要求为1且元素要求大于或等于0。 |
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/int64 |
ND |
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
y |
与传入的AsStridedParam中的size一致 |
float16/int64 |
ND |