码迷,mamicode.com
首页 > 其他好文 > 详细

Numpy:点积和 dot() 矩阵相乘

时间:2019-01-23 21:39:12      阅读:261      评论:0      收藏:0      [点我收藏+]

标签:bsp   ==   img   array   src   --   strong   sha   import   

一、Numpy的点积和dot矩阵相乘

dot()使用需注意:前一个矩阵的行数要与后一个矩阵的列数一致

import numpy as np
print("============点积:A*B,对应位置相乘===============")
A = np.array( [[1,1],
               [0,1]] )
B = np.array( [[2,0],
               [3,4]] )
print ("A=","\n",A)
print ("B=","\n",B)

print ("A*B=","\n",A*B)
# A.*B =[[1*2,1*0]
#        [0*3,1*4]]
#      =[[2,0]
#       [0,4]]

print("=============dot:矩阵相乘======================")

print("-----书写格式1:A.dot(B)-----")
print (A.dot(B))   
# A.dot(B)=[[1*2+1*3,1*0+1*4]
#           [0*2+1*3,0*0+1*4]]
#        =[[5,4]
#          [3,4]]


print("-----书写格式2:np.dot(A, B)-----")
print (np.dot(A, B)) 
# A.dot(B)=[[1*2+1*3,1*0+1*4]
#           [0*2+1*3,0*0+1*4]]
#         =[[5,4]
#           [3,4]]

结果图:

技术分享图片

 

 代码2:dot()使用需注意:前一个矩阵的行数要与后一个矩阵的列数一致

import numpy as np
a=np.ones((1,3,5))
print(a)
b=np.ones((5,6))*3
print(b)
c=a.dot(b)

print(c.shape)
print(c)

结果图:

技术分享图片

 

Numpy:点积和 dot() 矩阵相乘

标签:bsp   ==   img   array   src   --   strong   sha   import   

原文地址:https://www.cnblogs.com/wodexk/p/10311233.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!