Python NumPy expand_dims方法用法及代碼示例

Numpy 的 expand_dims(~) 方法返回一個新的 Numpy 數組，其中輸入數組的維度已擴展。

1. a | array-like

輸入數組。

2. axis | int

將添加新尺寸的軸。

一個新的 Numpy 數組，其中輸入數組的維度已擴展。

假設我們有以下一維數組：

a = np.array([1,2,3,4])
a



array([1, 2, 3, 4])

我們可以通過兩種方式將其擴展到二維：沿第 0 軸或第 1 軸添加新維度。

對於第 0 軸：

np.expand_dims(a, axis=0)



array([[1, 2, 3, 4]])

對於第 1 軸：

np.expand_dims(a, axis=1)



array([[1],
       [2],
       [3],
       [4]])

相關用法

注：本文由純淨天空篩選整理自Isshin Inada大神的英文原創作品 NumPy | expand_dims method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。