昇腾社区首页
中文
注册
开发者
下载

互推导关系

推导规则

当一个API(如aclnnAdd、aclnnMul等)输入的aclTensor数据类型不一致时,API内部会推导出一个数据类型,将输入数据转换成该数据类型进行计算。

aclTensor支持的数据类型参见,其中部分类型满足如下推导规则,推导原理类似PyTorch的

[object Object]

表 1 数据类型推导关系

[object Object]undefined

推导示例

  • 调用aclnnAdd接口时,如果输入参数的数据类型不一致,一个为float16,一个为float32,那么API内部就会将float16的数据类型转换成float32的数据类型然后进行计算。
  • 调用aclnnAdd接口时,如果输入参数的数据类型不一致,一个为float32,一个为bool,那么API内部就会将bool的数据类型转换成float32的数据类型然后进行计算。