添加不同形状python的matices

Question

我有2个不同尺寸的阵列。让我们使用numpy库用零填充它们：

A = np.zeros((24, 24, 24, 24))
B = np.zeros((24, 84, 24, 24))

我想添加这两个数组A + B以获得具有维度的新矩阵：

C = [ 24, 108, 24, 24]

我想过使用bradcasting和方法添加：

C = np.add(A, B)

但它不起作用。

你知道如何解决这个问题吗？

Answer 1

如果添加您的意思是追加，请尝试：np.append(a, b, axis = 1)。

assert np.append(a, b, axis = 1).shape == (24, 108, 24, 24)。