使用 numpy 組合兩個 n 維 arrays

Question

我有 2 個形狀為（2500、32、32、3）的 arrays。 我想將它們組合成一個數組 (5000, 32, 32, 3) 我該怎么做 go 呢？

Answer 1

使用 numpy 的 rowstack function。

c = np.row_stack((a,b))
print(c.shape)   # (5000, 32, 32, 3)