广播机制(broadcast)是一种重要机制,有了这种机制,就可以方便地对不同维的张量进行运算
比如:
c=np.arange(10).reshape([5,2])
d=np.arange(2).reshape([1,2])
c:
array([[0, 1],
[2, 3],
[4, 5],
[6, 7],
[8, 9]])
d:
array([[0, 1]])
c+d # 首先将d数组进行复制扩充为[5,2],然后相加。
运算结果:
array([[ 0, 2],
[ 2, 4],
[ 4, 6],
[ 6, 8],
[ 8, 10]])
原理: