CANN算子库TransDataSpecial

TransDataSpecial【免费下载链接】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数据的值实现对tensor数据格式的转换功能与TransData类似。具体功能是将用户输入tensor的format转换为指定的dstPrimaryFormat。函数原型const aclTensor *TransDataSpecial(const aclTensor *x, op::Format dstPrimaryFormat, int64_t groups, aclOpExecutor *executor)参数说明参数输入/输出说明x输入待转换的tensor。数据类型支持FLOAT16、FLOAT32、INT32、UINT32、INT8、UINT8。dstPrimaryFormat输入输入tensor要转换的目标format。groups输入分组参数用于分组转换时传入。数据类型支持INT64。executor输入op执行器包含了算子计算流程。返回值说明返回数据格式为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());【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考