MatMul
接口功能
执行两个矩阵的乘法运算。
函数原型
1 | int Matmul(T *blockA, T *blockB, T *output, MatMulParams<T> &M); |
参数说明
参数 |
输入/输出 |
说明 |
约束 |
---|---|---|---|
blockA |
输入 |
矩阵用于乘法运算。 |
指针不为空,且只接受一维张量。矩阵中不含有nan或者inf,需用户保障,防止计算结果未定义。 |
blockB |
输入 |
矩阵用于乘法运算。 |
指针不为空,且只接受一维张量。矩阵中不含有nan或者inf,需用户保障,防止计算结果未定义。 |
output |
输入 |
用于存储计算结果,构成的张量,作为输出。 |
指针不为空,且只接受一维张量。 |
M |
输入 |
用于乘法运算的参数结构体,包括:
|
父主题: 算子接口