如何將4維Numpy數組重塑為不同的尺寸？

Question

我有一個(8, 1, 1, 102) Numpy的4維Numpy數組。 現在，例如，我只是想忽略中間的兩個維度，並擁有一個形狀數組(8,102) ，什么是實現此目的的合適方法？

Answer 1

您不能簡單地“忽略”前兩個維度。 您有一個大小為8 * 1 * 1 * 102 == 816的數組，但是您想要一個大小為1 * 102的數組，因此必須選擇要刪除的值。

例如，如果您想要前102個，則可以執行

array[0, 0]

形狀將為(1, 102)

編輯

如果您要np.squeeze尺寸(8, 102) np.squeeze (8, 102)則就像刪除答案的其他用戶所說的那樣，您需要np.squeeze 。

x = np.random.random((8, 1, 1, 102))
y = np.squeeze(x)
print(y.shape)  # (8, 102)

Answer 2

np.squeeze將折疊所有長度為1的尺寸，或者您可以使用重塑功能