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矩陣的
相關文章
相關標籤/搜索