Transpose

支持的产品型号

Atlas 训练系列产品

Atlas A2训练系列产品/Atlas 800I A2推理产品

Atlas 推理系列产品

函数功能

该函数不改变tensor数据的值，而是对tensor进行转置操作。

通过transpose算子，可以改变tensor在不同维度上的排列顺序，实现对tensor的维度重排。

具体的是将用户传入的输入tensor x 的shape按指定维度的排列顺序perm进行转置并输出。

函数原型

• const aclTensor *Transpose(const aclTensor *x, const aclTensor *y, const aclTensor *perm, aclOpExecutor *executor)
• const aclTensor *Transpose(const aclTensor *x, const aclIntArray *perm, aclOpExecutor *executor)

参数说明

返回值说明

转置成功，返回转置后的tensor；转置失败，返回nullptr。

约束说明

1. 最多支持8维转置，即输入x和perm的dim至多为8

2. 输入x和perm的dim维度需要一致

3. 输入tensor x需是连续内存数据