ScalarCast
功能说明
将一个scalar的类型转换为指定的类型。
函数原型
template <typename srcT, typename dstT, RoundMode roundMode> __aicore__ inline dstT ScalarCast(srcT valueIn)
参数说明
| 参数名 | 输入/输出 | 描述 | ||
|---|---|---|---|---|
| valueIn | 输入 | 被转换数据类型的scalar。 | ||
| srcT | 输入 | valueIn的数据类型,支持float。 | ||
| dstT | 输入 | 转换后的数据类型,支持half、int32_t。 | ||
| roundMode | 输入 | 精度转换处理模式,类型是RoundMode。 RoundMode为枚举类型,用以控制精度转换处理模式,具体定义为: 
 对于ScalarCast,转换类型仅支持float转half(f322f16)与float转int32_t(f322s32),相应支持的RoundMode如下: f322f16:CAST_ODD f322s32:CAST_ROUND、CAST_CEIL、CAST_FLOOR、CAST_RINT ScalarCast的精度转换规则与Cast保持一致,具体可参考表1。 | 
返回值
dstT类型的valueIn。
支持的型号
Atlas推理系列产品AI Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
约束说明
无。
调用示例
| 1 2 3 | float valueIn = 2.5; // 输出数据(valueOut): 3 int32_t valueOut = AscendC::ScalarCast<float, int32_t, AscendC::RoundMode::CAST_ROUND>(valueIn); | 
父主题: 标量计算