ColwiseMul算子
算子概述
ColwiseMul(列方向复数乘法)算子实现了复数向量与矩阵每行的逐元素乘法运算,是 BLAS 基础线性代数库中的扩展算子之一。该算子针对复数运算特性进行了优化,高效完成向量与矩阵的列方向乘法操作。
数学表达式:
result[i, j] = vec[i] * mat[i, j]
包含以下接口:
| 接口名 | 功能简述 |
|---|---|
| aclblasColwiseMul | 复数向量与矩阵每行逐元素乘法 |
算子执行接口
aclblasColwiseMul
产品支持情况
- Ascend 950PR / Ascend 950DT:不支持
- Atlas A3 训练系列产品 / Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品 / Atlas A2 推理系列产品:支持
函数原型
int aclblasColwiseMul(const float *mat, const float *vec, float *result, const int64_t m, const int64_t n, void *stream);
参数说明
| 参数名 | 输入/输出 | 参数类型 | 说明 |
|---|---|---|---|
| mat | 输入 | const float*(FP32) | 复数矩阵,维度为 m x n,存储为 2mn 个 float,Device 内存 |
| vec | 输入 | const float*(FP32) | 复数向量,包含 m 个元素,存储为 2*m 个 float,Device 内存 |
| result | 输出 | float*(FP32) | 复数矩阵,维度为 m x n,存储为 2mn 个 float,Device 内存 |
| m | 输入 | int64_t | 矩阵的行数,Host 内存 |
| n | 输入 | int64_t | 矩阵的列数,Host 内存 |
| stream | 输入 | void* | 执行流,Host 内存 |
约束说明
无