Python numpy串聯4D

Question

我正在嘗試連接由vgg16模型中的卷積層預測的2個numpy特征數組。

基本上，我已經使用vgg16模型的底層來預測整個數據集的功能，現在我想基於某些設置動態加載數據集的各個部分，並以此來訓練一些模型。

因此，我有2個形狀數組： (724, 512, 6, 8)和(3376, 512, 6, 8)基本上第一個包含從724個圖像文件預測的特征（每個預測的形狀為(512, 6, 8) ）。 我想將這2個數組連接成一種形狀(4100, 512, 6, 8)

我試過使用：

np.array([np.concatenate(arr, axis=0) for arr in false_train_list])

其中false_train_list是包含上述形狀的2個數組的列表。

還嘗試了np.stack，tf.stack ...所有這些都導致形狀為（2，）的數組

有人可以解釋為什么嗎？ 我沒有找到任何好的資源來了解np.concatenate（）的工作原理。

謝謝！

Answer 1

我認為您只需要此即可：

np.concatenate(false_train_list, axis=0)