np.newaxis的功能:插入新維度 舉個簡單的例子介紹一下吧。

栗子1： a=np.array([1,2,3,4,5]) print(a.shape) print (a) 輸出： (5,) [1 2 3 4 5]

栗子2：

a=np.array([1,2,3,4,5])
aa=a[:,np.newaxis]
print(aa.shape)
print (aa)

輸出：(5, 1) [[1] [2] [3] [4] [5]]

栗子3：

a=np.array([1,2,3,4,5])
aa=a[np.newaxis,:]
print(aa.shape)
print (aa)

輸出 (1, 5) [[1 2 3 4 5]]

看明白了吧，原來np.newaxis的作用是增加一個維度。 對於[: , np.newaxis] 和 [np.newaxis，：] 是在np.newaxis這裡增加1維。

這樣改變維度的作用往往是將一維的資料轉變成一個矩陣，與程式碼後面的權重矩陣進行相乘， 否則單單的資料是不能呢這樣相乘的哦。