TransDataSpecial

支持的产品型号

Atlas 训练系列产品

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

Atlas 推理系列产品

函数功能

该函数不改变tensor数据的值，实现对tensor数据格式的转换，功能与TransData类似。

具体功能是将用户输入tensor的format转换为指定的dstPrimaryFormat。

函数原型

const aclTensor *TransDataSpecial(const aclTensor *x, op::Format dstPrimaryFormat, int64_t groups, aclOpExecutor *executor)

参数说明

返回值说明

返回数据格式为dstPrimaryFormat的tensor。

约束说明

当输入tensor数据类型为FLOAT32、INT32、UINT32时，C0只能按照16处理。

调用示例

// 固定写法，创建OpExecutor
auto uniqueExecutor = CREATE_EXECUTOR();
// 将gradOutputReFormat的format转换为NC1HWC0
auto gradOutputTransData = l0op::TransDataSpecial(gradOutputReFormat, op::Format::FORMAT_NC1HWC0, 0,uniqueExecutor.get());