linear-algebra - BLAS:gemm 与 gemv

Question

为什么 BLAS 有 gemm矩阵-矩阵乘法函数和一个单独的gemv矩阵向量乘法的函数？矩阵向量乘法不只是矩阵矩阵乘法的一种特殊情况，其中一个矩阵只有一行/列吗？

Answer 1

在数学上，矩阵-向量乘法是矩阵-矩阵乘法的一种特殊情况，但在软件库中实现的不一定是这样的。

他们支持不同的选择。例如，gemv支持跨步访问它正在操作的向量，而 gemm不支持跨步矩阵布局。在 C 语言绑定(bind)中，gemm要求您指定所有三个矩阵的存储顺序，而在 gemv 中这是不必要的。对于向量参数，因为它没有意义。

除了支持不同的选项外，还有一些可以在 gemm 上执行的优化系列。不适用于 gemv .如果您知道自己正在做矩阵向量乘积，那么您不希望库在切换到针对该情况优化的代码路径之前浪费时间弄清楚情况；你宁愿直接调用它。