一文浅析np.newaxis

---

前言

np.newaxis的功能 —— 插入新维度

---

一、栗子

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

---

当使用[: , np.newaxis] 后，代码为：

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

---

当使用[np.newaxis,:] 后，代码为：

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

二、总结

使用np.newaxis来改变矩阵的维度，使之与代码后面的权重矩阵进行相乘。