加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 编程开发 > Python > 正文

python – BLAS sgemm / dgemm如何工作?

发布时间:2020-12-20 11:18:22 所属栏目:Python 来源:网络整理
导读:我试图在 python中使用ctypes在BLAS中使用函数sgemm.试图解决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
我试图在 python中使用ctypes在BLAS中使用函数sgemm.试图解决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,B,byref(zero),C,byref(m))

现在我想解决这个等式:C = A’x A其中A’是A的转置,下面的代码运行时没有异常,但返回的结果是错误的:

trans = c_char("t")
no_trans = c_char("n")
m = c_int(number_of_rows_of_A)
n = c_int(number_of_columns_of_A)
one = c_float(1.0)
zero = c_float(0.0)

blaslib.sgemm_(byref(trans),byref(n))

对于测试我插入了一个矩阵A = [1 2; 3 4].正确的结果是C = [10 14;但是sgemm例程吐出C = [5 11; 11 25].

据我所知,矩阵A不必由我调换,因为算法会处理它.在第二种情况下我的参数传递出了什么问题?

任何帮助,链接,文章,建议表示赞赏!

解决方法

Blas通常使用列主矩阵(如Fortran),因此A = [1 2; 3 4]意思

|1 3|   
A = |   |
    |2 4|

结果是正确的(假设你的Python库也是这样).见这read-me

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读