把数组铺开,扩展之后无法匹配就无法进行广播。
可以广播的条件:
1.数组维度不同,轴长度相同
a =np.array( [ [1,2,3] , [3,2,1] ] )
b= np.array( [ [1,1,1] ] )
print(a+b)
output:
[[2 3 4]
[4 3 2]]
2.数组维度相同,且有个轴长度为1或者0
a =np.array( [ [1,2,3] , [3,2,1] ] )
b= np.array( [ [2] , [3] ] )
print(a+b)
ouput:
[[3 4 5]
[6 5 4]]
网友评论