添加两个三维 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)
我有两个 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)