Cast【免费下载链接】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转换为指定的数据类型。函数原型const aclTensor *Cast(const aclTensor *self, op::DataType dstDtype, aclOpExecutor *executor)参数说明参数名输入/输出说明self输入待转换的输入tensor数据类型支持FLOAT16、FLOAT、DOUBLE、BFLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL、COMPLEX64、COMPLEX128。数据格式支持ND。dstDtype输入转换后的目标dtype数据类型支持FLOAT16、FLOAT、DOUBLE、BFLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL、COMPLEX64、COMPLEX128。executor输入op执行器包含了算子计算流程。说明BFLOAT16仅适用于如下产品Atlas A2 训练系列产品/Atlas A2 推理系列产品Atlas A3 训练系列产品/Atlas A3 推理系列产品返回值说明返回类型为dstDtype的tensor。约束说明无调用示例// 标准写法创建OpExecutor auto uniqueExecutor CREATE_EXECUTOR(); auto selfCasted self; // 当self为布尔类型时利用Cast接口转换为uint8类型后可进行整型计算 if (self-GetDataType() op::DataType::DT_BOOL) { selfCasted l0op::Cast(self, op::DataType::DT_UINT8, uniqueExecutor.get()); CHECK_RET(selfCasted ! nullptr, ACLNN_ERR_PARAM_NULLPTR); }【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考