ComplexMatDot
功能描述
计算公式
接口原型
- AspbStatus asdBlasMakeComplexMatDotPlan(asdBlasHandle handle)
- AspbStatus asdBlasComplexMatDot(asdBlasHandle handle, const int64_t m, const int64_t n, aclTensor *matx, aclTensor *maty,aclTensor *result);
参数列表
- asdBlasMakeComplexMatDotPlan
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
ComplexMatDot算子的句柄。
- asdBlasComplexMatDot
规格约束
- asdBlasMakeComplexMatDotPlan:无。
- asdBlasComplexMatDot
- 非原地更新情况下,输入和输出占用的内存大小应不超过NPU的GM大小;
- 原地更新情况下,输入占用的内存大小应不超过NPU的GM大小;
- 算子输入shape为[m,n]、[m,n],输出shape为[m,n];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
父主题: BLAS