numpy reshape 不能從 (a,b,c) 到 (a,b,c,1)？

Question

我將張量從 (a,b,c) 重塑為 (a,b,c,1)，但結果是 (a,1,b,c) 使用 numpy。有人知道為什么嗎？ 或者有什么方法可以從 (a,b,c) 得到 (a,b,c,1)？

Answer 1

您可以使用

your_array = np.expand_dims(your_array, axis=3)

Answer 2

您可以簡單地將 1 添加到arr.shape元組：

arr = np.arange(20).reshape(5,2,2)
arr.shape += (1,)
print(arr.shape)

Output：

(5, 2, 2, 1)

Answer 3

我建議使用 np.newaxis ，如下例所示：

tensor1 = np.array([[[1,2],[3,4],[5,6]],[[7,8],[9,10],[11,12]]])
print(tensor1.shape)

output：（2、3、2）

tensor2 = tensor1[..., np.newaxis]
print(tensor2.shape)

output：（2、3、2、1）