kutacc_af2_linear
linear线性变化的kutacc实现, 对应公式:
接口定义
void kutacc_af2_linear(kutacc_tensor_h act, kutacc_tensor_h weight, float* bias_data, kutacc_tensor_h result, int64_t beta);
参数
参数名 |
类型 |
描述 |
输入/输出 |
|---|---|---|---|
act |
kutacc_tensor_h |
基础数据A |
输入 |
weight |
kutacc_tensor_h |
权重参数B |
输入 |
bias_data |
float * |
偏移量参数 |
输入 |
result |
kutacc_tensor_h |
运算结果 |
输出 |
beta |
int64_t |
C的系数 |
输入 |
参数约束关系:
bias维度大小等于weight维度大小减1，
act & weight维度均大于等于2
父主题： GEMM算子