Why does numpy Dot product of 2d array with 1d array produce 1d array?

前端 未结 4 1052
感情败类
感情败类 2020-12-22 01:07

I try to run the code like below:

>>> import numpy as np
>>> A = np.array([[1,2], [3,4], [5,6]])
>>> A.shape
(3, 2)
>>> B         


        
4条回答
  •  暗喜
    暗喜 (楼主)
    2020-12-22 01:53

    In wiki

    So (3, 2) dot with (2,1) will be (3,1)


    How to fix

    np.dot(A,B[:,None])
    Out[49]: 
    array([[23],
           [53],
           [83]])
    

提交回复
热议问题