在 numpy 中做到這一點的簡單方法？

Question

假設我有一個 BxNxL 數組 M。換句話說，M 由 B NxL 個矩陣組成。 此外，我有一個 LxB 列向量 Q。是否有任何簡單的方法（沒有 for 循環）將 Q 的第 i 列廣播（求和）到 M 中的第 i 個矩陣？

Answer 1

所以你的迭代代碼會是？

for i in range(...):    
    res[i,:,:] = M[i,:,:] + Q[:,i]    # NxL + L

與整個陣列

res = M + Q.T[:,None,:]     # BxNxL  + (Bx1xL)

（我寫的沒有測試例子，所以可能有錯誤，但基本思路應該是對的）