添加两个三维 numpy 数组:

adding two three dimensional numpy array:

我有两个 numpy 数组:X 形状是 (68,44,13) 和 X_toadd 形状是:(68,44,7)我想以我想要的方式将它们加在一起X_new 形状为 (68,44, 20)。因此,我需要保留 X 的前两个维度并将 X_toadd 的第三个维度的 7 列添加到 13 列。 我该怎么做?

尝试了添加、附加和连接,但结果不是我想要的,应该具有 (68,44,20) 的形状!

您需要指定使用哪个轴将事物粘合在一起。 这里,-1表示从后数第一个轴。

import numpy as np
a,b = np.zeros((68,44,13)), np.zeros((68,44,7))

c = np.concatenate([a,b], axis=-1)

c.shape
(68, 44, 20)