二維數組與3D數組相乘

Question

我有 3D A 矩陣 3x3x5（第三維是 5）和 2D B 矩陣（3x3）。 我想將 A 和 B 相乘以獲得 (3x3x5) 矩陣。 然后將生成的 3D 矩陣的元素相加以創建二維矩陣 (3x3)。 我怎樣才能做到這一點？

Answer 1

只需使用*運算符將 numpy 與 arrays 相乘即可。

import numpy as np
a = np.arange(45).reshape(3, 3, 5)
b = np.arange(9).reshape(3, 3)
c = a * b
print(c) # 3x3x5 array
d = np.sum(c, axis=-1)
print(d)

d應該是您正在尋找的答案。