TransposeOperation
Description
Rearranges the input tensors and transposes them in multiple dimensions.
Figure 1 TransposeOperation
Definition
struct TransposeParam {
SVector<int32_t> perm;
uint8_t rsv[8] = {0};
};
Parameters
Member |
Type |
Default Value |
Description |
|---|---|---|---|
perm |
SVector<int32_t> |
- |
Rearrangement result of the input dimensions. Ensure that the input of perm is correct and has the same dimension as input x. |
rsv[8] |
uint8_t |
{0} |
Reserved |
Input
Parameter |
Dimension |
Data Type |
Format |
|---|---|---|---|
x |
[-1,…,-1] |
float16/bf16/int64 |
ND |
Output
Parameter |
Dimension |
Data Type |
Format |
|---|---|---|---|
output |
[-1,…,-1] |
float16/bf16/int64 The data type is the same as that of x. |
ND |
Parent topic: atb/infer_op_params.h