matrix-multiplication - 英特尔 MKL 矩阵产品 gemm 是否就地工作？

Question

英特尔 MKL 中的 BLAS gemm 例程通常使用三个矩阵。这就像告诉你f(A, B, C) = alpha A * B + beta Calpha 和 beta 是比例因子。

但是为了简单地得到 A = A * B可以写f(A, B, A)吗？alpha=1,beta=0我的意思是两个Ainf(A, B, A)是同一个变量。（这里所有的方阵。）

当然，如果我们设置第三个变量C=A，就f(A, B, C)可以了。但是即使不制作这个副本也会好得多C。

标签： matrix-multiplicationblasintel-mkl

Answer 1

不，这是不允许的。您要么必须引入一个临时缓冲区，要么找到另一种方法来完成这项工作。