AI硬體與軟體堆疊
計算M×K矩陣與K×N矩陣的乘積時,浮點運算的次數(乘法與加法分開計算)大約是多少?
- AMNK
- BM+N+K
- C2MNK
查看答案與解析
正確答案:C. 2MNK
結果矩陣有M×N個元素,每個元素是長度為K的內積,因此需要K次乘法與K次加法(約K次)。合計約為2MNK次運算。AI的計算量常以此式估算,效能則以每秒的運算次數(FLOPS)表示。若將乘加視為1次,則會減半為MNK,因此要注意對方是以哪一種方式計算。
- A. 這是將乘加視為1次時的次數。
- B. 運算次數會多得多。
- C. 正確。每個元素各進行K次乘法與加法。