numpy：形状向量（5,1）和（5，）有什么区别？[副本]

Question

我发现为了使用点积，我经常不得不将（5，）重塑为（5,1）。我不能使用形状向量（5，）的点积吗？

Answer 1

这是因为当您使用arr=np.ones（（5））创建一个数组时，它将得到一个包含5个元素的1D数组，另一方面，当您使用arr=np.ones（（5，1））创建数组时，它将创建一个包含5行1列的2D数组。下面的示例会让您更清楚：

>>> import numpy as np 
>>> a = np.ones((5, 1))
>>> a
array([[ 1.],
       [ 1.],
       [ 1.],
       [ 1.],
       [ 1.]])
>>> a = np.ones((5))
>>> a
array([ 1.,  1.,  1.,  1.,  1.])

Answer 2

要使用点积，需要矩阵（用2D数组表示）。维度为（5，）的数组是由5项组成的平面数组（1D数组），其中as（5，1）是包含1列和5行的矩阵。

>>> import numpy as np
>>> np.zeros((5,))
array([ 0.,  0.,  0.,  0.,  0.])    # single flat array
>>> np.zeros((1,5))
array([[ 0.,  0.,  0.,  0.,  0.]])  # array with-in array
>>> np.zeros((5,1))
array([[ 0.],
       [ 0.],
       [ 0.],
       [ 0.],
       [ 0.]])
>>>