TransData

支持的产品型号

Atlas 训练系列产品

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

Atlas 推理系列产品

函数功能

该函数不改变tensor数据的值，实现对tensor数据格式的转换。

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

函数原型

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

参数说明

返回值说明

返回数据格式为dstPrimaryFormat的tensor。

约束说明

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

调用示例

// 将输出的格式从NC1HWC0转换成NCHW
auto transGradInput = l0op::TransData(gradInputNC1HWC0, Format::FORMAT_NCHW, params.groups, executor);
CHECK_RET(transGradInput != nullptr, ACLNN_ERR_INNER_NULLPTR);