功能
改变输入Tensor的排列顺序，在多个维度上进行转置。
图1 TransposeOperation
定义
struct TransposeParam {
SVector<int32_t> perm;
};
成员
成员名称 |
描述 |
|---|---|
perm |
指示输入维度的重排结果。 perm需要保证输入正确，维度和输入x一致。 |
输入
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/int64/bf16 |
ND |
输出
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
output |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/int64/bf16 |
ND |