Cast
贡献说明
| 贡献者 | 贡献方 | 贡献算子 | 贡献时间 | 贡献内容 |
|---|---|---|---|---|
| skywang2 | 个人开发者 | Cast | 2025/12/31 | 新增Cast算子 |
算子描述
Cast算子提供将tensor从源数据类型转换为目标数据类型的功能。
算子规格描述
| 算子类型(OpType) | Cast | |||
|---|---|---|---|---|
| 算子输入 | name | type | data type | format |
| x | tensor | float16, float32, int32, int8, uint8, bool, int64, bfloat16, int16 | ND | |
| 算子输出 | out | tensor | float16, float32, int32, int8, uint8, bool, int64, bfloat16, int16 | ND |
| 算子属性 | dstType | attr | int64 | - |
| 核函数名 | cast | |||
支持的产品型号
本样例支持如下产品型号：
- Atlas A2训练系列产品
- Atlas 800I A2推理产品
环境要求
编译运行此样例前，请参考《CANN软件安装指南》完成开发运行环境的部署。
算子调用
测试命令调用方式：build.sh
| 目录 | 描述 |
|---|---|
| test_aclnn_cast.cpp | 通过aclnn调用的方式调用Cast算子 |