我本来想计算一个C=alphaAB,其中A[3][5]={{1,2,3,4,5},{6,7,8,9,10},{11,12,13,14,15}};B[5][3]={{1,2,3},{2,4,6},{3,6,9},{4,8,12},{5,10,15}};alpha=5.0;
我知道cublasSgenmm()中矩阵是按照列为主进行储存的,所以我得写成A[5][3]={{1,2,3},{2,4,6},{3,6,9},{4,8,12},{5,10,15}};B[3][5]={{1,2,3,4,5},{6,7,8,9,10},{11,12,13,14,15}};cublasSgemm(handle,transa,transb,3,3,5,&alpha,dev_A,3,dev_B,5,&beta,dev_C,3);其中beta=0;
感觉这样用起来好麻烦呀,我还得把A,B矩阵进行修改,有没有一种办法直接输入A,B矩阵就能直接计算呢?