[問題] 多個矩陣做點對點乘法再相加怎樣最快

看板Python作者doasgloria (青柳立夏)時間5年前 (2020/04/07 03:07)推噓4(4推 0噓 4→)

留言8則, 6人參與, 5年前最新討論串1/1

現在想要做的是假設有矩陣 A,B,C,...J共約10個要與同樣數量shape的矩陣a,b,c,...,j 每個shape大約是 (3500,2500,3) 做 A*a + B*b + C*c +...+ J*j 點對點相乘再全部加總所以出來的結果也是一個(3500,2500,3)的矩陣該怎麼做最快呢 @_@" type都是numpy array,嘗試了幾種方法 1. 直接使用np multiply方法再相+ 當然這是最慢的, 要好多秒 2. 使用numexpr把"A*a+B*b+.."的算式寫在evaluate裡面這個很強快超多大概30%以上但還是不夠快 3. 使用Cupy 這個當然是最快的可是存在一個問題我把20個矩陣用cp.array()轉到cupy 計算完後用cp.asnumpy()轉回來這兩個過程滿耗時的加起來又比上面兩個慢了 Q.Q 目前嘗試了上面三種方法都沒辦法達到理想的速度假設在 i5-7400CPU , 16G記憶體 , 1050Ti的GPU上有方法可以達到一秒鐘計算50次甚至100次嗎 ?? 希望版上的計算達人們指教還是小蛇的小魯 >///< -- 　　 ◤ | █　　　ヽ █ ◥　　　 ▂▃▄▃▂ 回戰文疾如風　　 ▍˙▆ 八███████▃▃▄　 ●火 ◇ 風● 　發廢文繁如林　　▋˙●◤█◤ノ　　　ヽ◥ ‧▋ 　 ●◇ ◇● 挑釁文嗆如火　　 ◥___ █◤ ●　　 ● ██◣ 　 ●山◇林● 宅在家不動如山　　 ◢█ ██◤　　( _●_) ミ█◣ 彡　　甲斐の熊－武田熊玄 ◢▇▇▊█▼ 　　|∪|　 ◢█ ◣_█◢◢ ◤、● 原SotaFujimor 改marubom -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.45.58.106 (臺灣) ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1586228873.A.1CC.html

推

TuCH

04/07 12:12, 5年前 , 1^F

04/07 12:12, 1^F

→

doasgloria

04/07 12:36, 5年前 , 2^F

04/07 12:36, 2^F

→

doasgloria

04/07 12:37, 5年前 , 3^F

04/07 12:37, 3^F