脾氣暴躁的廣播

Question

當我在Numpy中執行此操作時會發生什么？

a = np.ones([500,1])
b = np.ones([5000,])/2
c = a + b 

# a.shape (500,1)
# b.shape (5000, )
# c.shape (500, 5000)

我很難弄清楚此廣播中實際發生的情況。

Answer 1

Numpy假定一維數組為行向量，因此您的求和確實在形狀（500，1）和（1，5000）之間，這導致矩陣求和。

由於這不是很清楚，因此您應該顯式擴展尺寸：

>>> np.arange(5)[:, None] + np.arange(8)[None, :]
array([[ 0,  1,  2,  3,  4,  5,  6,  7],
       [ 1,  2,  3,  4,  5,  6,  7,  8],
       [ 2,  3,  4,  5,  6,  7,  8,  9],
       [ 3,  4,  5,  6,  7,  8,  9, 10],
       [ 4,  5,  6,  7,  8,  9, 10, 11]])