SetBias
功能说明
设置矩阵乘的Bias。
函数原型
__aicore__ inline void SetBias(GlobalTensor<T>& biasGlobal);
__aicore__ inline void SetBias(LocalTensor<T>& biasLocal);
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
biasGlobal |
输入 |
Bias在Global Memory上的首地址，类型为GlobalTensor。 Atlas推理系列产品AI Core，支持的数据类型为：float Atlas A2训练系列产品，支持的数据类型为：half/float |
biasLocal |
输入 |
Bias在VECCALC/TSCM上的首地址，类型为LocalTensor。 Atlas推理系列产品AI Core，支持的数据类型为：float Atlas A2训练系列产品，支持的数据类型为：half/float |
返回值
无
支持的型号
Atlas A2训练系列产品
Atlas推理系列产品AI Core
注意事项
传入的Bias地址空间大小需要保证不小于singleN。
父主题： Matmul 高阶API