前言
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来改变矩阵的维度,使之与代码后面的权重矩阵进行相乘。