ReFormat

支持的产品型号

Atlas 训练系列产品

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

Atlas 推理系列产品

函数功能

该函数不改变tensor数据的值，在指定format和输入x的维度相同时，将输入数据格式设置为目标format。

具体功能是将用户输入tensor的viewFormat、originalFormat、storageFormat统一为指定的format。

函数原型

const aclTensor *ReFormat(const aclTensor *x, const op::Format &format, aclOpExecutor *executor=nullptr)

参数说明

返回值说明

返回数据格式为目标format后的tensor。

约束说明

输入tensor的数据的维度需要与指定format的维度相同。

调用示例

// 将输入reformat成NCHW格式
auto reformatInput = l0op::ReFormat(unsqueezedInput, Format::FORMAT_NCHW);
CHECK_RET(reformatInput != nullptr, nullptr);