python 中numpy dot函数的使用方法


 这个函数在的数字信号处理中用处还是比较广泛的,函数的具体定义如下所示:

numpy.dot(a, b, out=None)

  该函数的作用是获取两个元素a,b的乘积,表示的含义如下所示:

dot(a, b)[i,j,k,m] = sum(a[i,j,:] * b[k,:,m])

  使用方法如下所示:

单个数:

>>> np.dot(3, 4)
12

复数:

>>> np.dot([2j, 3j], [2j, 3j])
(-13+0j)

二维矩阵:

>>> a = [[1, 0], [0, 1]]
>>> b = [[4, 1], [2, 2]]
>>> np.dot(a, b)
array([[4, 1],
       [2, 2]])

 

>>> a = np.arange(3*4*5*6).reshape((3,4,5,6))
>>> b = np.arange(3*4*5*6)[::-1].reshape((5,4,6,3))
>>> np.dot(a, b)[2,3,2,1,2,2]
499128
>>> sum(a[2,3,2,:] * b[1,2,:,2])
499128

  参考文档:

1 https://docs.scipy.org/doc/numpy/reference/generated/numpy.dot.html


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM