CANN算子库ReFormat函数文档

发布时间:2026/7/5 18:40:50
CANN算子库ReFormat函数文档 ReFormat【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs产品支持情况Ascend 950PR/Ascend 950DT 不支持Atlas A3 训练系列产品/Atlas A3 推理系列产品 支持Atlas A2 训练系列产品/Atlas A2 推理系列产品 支持Atlas 200I/500 A2 推理产品 不支持Atlas 推理系列产品 支持Atlas 训练系列产品 支持功能说明该函数不改变tensor数据的值在指定format和输入x的维度相同时将输入数据格式设置为目标format。具体功能是将用户输入tensor的viewFormat、originalFormat、storageFormat统一为指定的format。函数原型const aclTensor *ReFormat(const aclTensor *x, const op::Format format, aclOpExecutor *executornullptr)参数说明参数输入/输出说明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);【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考