cublasSgemm函数实现矩阵乘

①cublas中矩阵是按照列主元素存储的,因此想要计算A*B,那么输入的应该是B*A,就是把原来的B放在A的位置,把原来的A放在B的位置,这样子输出的就是A*B的结果。由于咱们输入B,其实cublas读入的就是B的转置,A就是A的转置,而后BT*AT = (AB)T,可是再按列主元读出来的时候就是AB啦。html ②m,n,k,以及lda,ldb,ldc这6个参数怎么输入啦。m是op(放在A矩阵的
相关文章
相关标签/搜索