下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

Transpose

支持的产品型号

Atlas 训练系列产品

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

Atlas 推理系列产品(Ascend 310P处理器)

函数功能

该函数不改变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)

参数说明

参数

输入/输出

说明

x

输入/输出

原始输入tensor。输入tensor x需是连续内存数据。数据类型支持FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL、BFLOAT16。数据格式支持ND。

说明:

其中BFLOAT16适用于如下产品型号:

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

y

输出

转置后输出tensor。数据类型和数据格式同x。

perm

输入

指定维度的排列顺序,支持aclIntArray*、aclTensor*类型。最多支持8维转置。数据类型支持INT32、INT64。数据格式支持ND。

executor

输入

op执行器,包含了算子计算流程。

返回值说明

转置成功,返回转置后的tensor;转置失败,返回nullptr。

约束说明

  • 最多支持8维转置,即输入x和perm的dim至多为8。
  • 输入x和perm的dim维度需要一致。

调用示例

// 固定写法,创建OpExecutor,参数检查
auto uniqueExecutor = CREATE_EXECUTOR();
CHECK_RET(uniqueExecutor.get() != nullptr, ACLNN_ERR_INNER_CREATE_EXECUTOR);

// 固定写法,将输入self转换成连续的tensor
auto selfContiguous = l0op::Contiguous(self, uniqueExecutor.get());
CHECK_RET(selfContiguous != nullptr, ACLNN_ERR_INNER_NULLPTR);

int64_t dims = selfContiguous->GetViewShape().GetDimNum();
int64_t valuePerm[dims] = {0, 2, 1, 4};

auto perm = executor->AllocIntArray(valuePerm, dims);
selfContiguous = l0op::Transpose(selfContiguous, perm, uniqueExecutor.get());
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词