python - numpy block 矩阵如何工作？

Question

这段代码的结果对我来说没有任何意义:

a = np.zeros((2, 2))
b = np.bmat([[a, a], [a, a]])
print(b.shape, b.dot(np.zeros(4)).shape)

形状为 (4, 4) 的矩阵在其最终轴上进行和积时如何返回形状为 (1, 4) 的矩阵？

Answer 1

bmat 返回一个 numpy.matrix 实例，因为在那些你不应该使用的东西中，因为它们会导致各种奇怪的不兼容性。 numpy.matrix 始终尝试保留至少两个维度，因此 b.dot(np.zeros(4)) 是二维的而不是一维的。

制作一个numpy.array:

b = np.bmat([[a, a], [a, a]]).A
#                             ^

或者从 NumPy 1.13 开始，

b = np.block([[a, a], [a, a]])