我试图利用BLAS中的函数sgemm
,使用python中的ctypes。尝试求解C=A x B以下代码可以正常工作:
no_trans = c_char("n")
m = c_int(number_of_rows_of_A)
n = c_int(number_of_columns_of_B)
k = c_int(number_of_columns_of_A)
one = c_float(1.0)
zero = c_float(0.0)
blaslib.sgemm_(byref(no_trans), byref(no_trans), byref(m), byref(n), byref(k),
byref(one), A, byref(m), B, byref(k), byref(zero), C, byref(m))
现在我想解这个方程:C=A'xa,其中A'是A的转置,下面的代码运行正常,但返回的结果是错误的:
^{pr2}$对于一个测试,我插入了一个矩阵a=[12;34]。正确的结果是C=[1014;1420],但是sgemm
例程会抛出C=[511;1125]。在
据我所知,矩阵A不必被我转置,因为算法会处理它。在第二种情况下传递参数有什么问题?在
任何帮助,链接,文章,建议是感激!在
Blas通常使用列主矩阵(如Fortran),因此
A = [1 2; 3 4]
表示结果是正确的(假设您的Python库也这样做)。看这个read-me
您得到的结果表明
sgemm
计算了一个*A'而不是一个您想要的'*A'。简单的解决方案是将两个输入切换到函数。在相关问题 更多 >
编程相关推荐