改变输入Tensor的排列顺序,在多个维度上进行转置。
1 2 3 4 | struct TransposeParam { SVector<int32_t> perm; uint8_t rsv[8] = {0}; }; |
成员名称 |
类型 |
默认值 |
描述 |
---|---|---|---|
perm |
SVector<int32_t> |
- |
指示输入维度的重排结果。 例如:输入tensor的shape是[a,b,c],perm是[0,2,1],表明交换第一维和第二维,输出tensor的shape是[a,c,b]。 perm需要保证输入正确,维度和输入x一致。 |
rsv[8] |
uint8_t |
{0} |
预留参数。 |
参数 |
维度 |
数据类型 |
格式 |
---|---|---|---|
x |
[dim_0,dim_1,... ,dim_n] |
float16/bf16/int64 |
ND |
参数 |
维度 |
数据类型 |
格式 |
---|---|---|---|
output |
维度由参数确定 |
float16/bf16/int64 数据类型与x保持一致 |
ND |