Colwise_mul
功能描述
计算公式
接口原型
- AspbStatus asdBlasMakeColwiseMulPlan(asdBlasHandle handle)
- AspbStatus asdBlasColwiseMul(asdBlasHandle handle, const int64_t m, const int64_t n, aclTensor *mat, aclTensor *vec,aclTensor *result)
参数列表
- asdBlasMakeColwiseMulPlan
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
ColwiseMul算子的句柄。
- asdBlasColwiseMul
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
ColwiseMul算子的句柄。
m
Input
const int64_t
矩阵mat的行数,向量vec的元素个数。
n
Input
const int64_t
矩阵mat的列数。
mat
Input
aclTensor *
公式中的A,Device侧的tensor,数据类型支持COMPLEX64 ,数据格式支持ND,shape为[m,n]。
vec
Input
aclTensor *
公式中的x,复数向量,Device侧的tensor,数据类型支持COMPLEX64 ,数据格式支持ND,shape为[m]。
result
Output
aclTensor *
公式中的result,Device侧的tensor,数据类型支持COMPLEX64,数据格式支持ND,shape为[m,n],数据格式、类型、shape与入参mat一致。
规格约束
- asdBlasMakeColwiseMulPlan:无。
- asdBlasColwiseMul
- 算子输入shape为[m,n]、[m],输出shape为[m,n];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
父主题: BLAS