numpy的广播(broadcast)机制

首先,广播只适用于从维度低(low rank)的ndarry(或rank=2的行向量或列向量)与维度高(high rank)的ndarry相加(或其他element-wise操作)才会有效。如果是同rank,则不存在广播机制,系统会报错。
如:

z = np.random.randint(1,4,(4,2)) 
# array([[2, 1],
#           [3, 2],
#           [3, 2],
#            [2,2]])
a = np.random.randint(1,4,(2,2))
# array([[1, 2],[2,3])
z+a

得出的结果是

ValueError                                Traceback (most recent call last)
<ipython-input-63-d213f4463bc1> in <module>()
      1 z = np.random.randint(1,4,(2,4))
      2 a = np.random.randint(1,4,(2,2))
----> 3 a+z

ValueError: operands could not be broadcast together with shapes (2,2) (2,4) 

下面来看看有效的广播操作,即rank-i+rank-j(i≤j)

z = np.random.randint(1,4,(4,3))
print(z)
a = np.array([1,2,3]) #或者令a=np.array([[1,2,3]]) rank-2也行。
a+z

结果为

[[1 1 2]
 [2 1 3]
 [3 1 3]
 [3 1 2]]
array([[2, 3, 5],
       [3, 3, 6],
       [4, 3, 6],
       [4, 3, 5]])

首先确定a.shape=(4,3),c.shape=(3,),然后C-->(1,3)-->(4,3)=
[[1, 2, 3],
[1, 2, 3],
[1, 2, 3],
[1, 2, 3]]

再来看rank-2+rank-3的广播情况。

z = np.random.randint(1,5,(4,4,3))
a = np.random.randint(1,4,(4,3))
z+a

其中z为

a为

z+a为

注意不要被这个(4,4,3)的样子给弄混淆,把它当做RGB图像来看,z[:,:,0]即图片的第一个通道,z[:,:,1]即图片的第二个通道,以此类推。加上a后,每个图片的第i个通道的每一行加上a的第i列(加的时候要转置)。

直观来看


©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 基础篇NumPy的主要对象是同种元素的多维数组。这是一个所有的元素都是一种类型、通过一个正整数元组索引的元素表格(...
    oyan99阅读 5,156评论 0 18
  • TF API数学计算tf...... :math(1)刚开始先给一个运行实例。tf是基于图(Graph)的计算系统...
    MachineLP阅读 3,552评论 0 1
  • NumPy是Python中关于科学计算的一个类库,在这里简单介绍一下。 来源:https://docs.scipy...
    灰太狼_black阅读 1,251评论 0 5
  • 先决条件 在阅读这个教程之前,你多少需要知道点python。如果你想从新回忆下,请看看Python Tutoria...
    舒map阅读 2,602评论 1 13
  • 纽约大学教授尼尔·波兹曼在其反乌托邦著作《娱乐至死》中曾开宗明义地提到过,“毁掉我们的,不是我们所憎恶的东西,而恰...
    清虚桂子飘阅读 8,990评论 7 20