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

ReFormat

支持的产品型号

Atlas 训练系列产品

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

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

函数功能

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

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

函数原型

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

参数说明

参数

输入/输出

说明

x

输入

需要被转换的tensor。数据类型支持FLOAT16、FLOAT32、INT32、UINT32、INT8、UINT8。

format

输入

输入tensor要转换的目标format。

executor

输入

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

返回值说明

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

约束说明

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

调用示例

// 将输入reformat成NCHW格式
auto reformatInput = l0op::ReFormat(unsqueezedInput, Format::FORMAT_NCHW);
CHECK_RET(reformatInput != nullptr, nullptr);
搜索结果
找到“0”个结果

当前产品无相关内容

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