TransposeOperation 功能改变输入Tensor的排列顺序,在多个维度上进行转置。 图1 TransposeOperation 定义struct TransposeParam { SVector<int32_t> perm; }; 成员 成员名称 描述 perm 指示输入维度的重排结果。 perm需要保证输入正确,维度和输入x一致。 输入 参数 维度 数据类型 格式 x [-1,…,-1] float16/bfloat16/int64 ND 输出 参数 维度 数据类型 格式 output [-1,…,-1] float16/bfloat16/int64 ND 父主题: atb/infer_op_params.h