Python的numpy中的 broadcasting(广播)机制

broadcasting,广播,传递,赋值,拷贝;
一定要注意,执行 broadcast 的前提在于,两个 ndarray 执行的是 element-wise(按位加,按位减) 的运算,而不是矩阵乘法的运算,矩阵乘法运算时需要维度之间严格匹配。(且矩阵乘法,np.dot(A, B) 如果维度不匹配,提示的错误不会是 broadcast,而是 aligned
我们常常会看到 python 编译器会提示如下类型的错误:

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

那么如何理解这里的broadcast呢,matlab中并无对等的概念?
broadcasting机制的功能是为了方便不同shape的array(numpy库的核心数据结构)进行数学运算。
举一个简单的例子,实现对一个1-d array的每一个元素乘以2:

>>> a = np.array([1., 2., 3.])
>>> b = np.array([2., 2., 2.])
>>> a*b
array([2., 4., 6.])

broadcast的做法是:

>>> a = np.array([1., 2., 3.])
>>> b = 2.
>>> a*b
array([2., 4., 6.])

这也就解释了之前常常令人困惑的(3, )不同于(3, 1)(表shape)。
我们来看更为一般的broadcasting rules:
当操作两个array时,numpy会逐个比较它们的shape(构成的元组tuple),只有在下述情况下,两arrays才算兼容:
1.相等
2.其中一个为(进而可进行拷贝拓展已至,shape匹配)

1. 举例
举例说明:

Image (3d array):  256 x 256 x 3
Scale (1d array):              3
Result (3d array): 256 x 256 x 3

A      (4d array):  8 x 1 x 6 x 1
B      (3d array):      7 x 1 x 5
Result (4d array):  8 x 7 x 6 x 5

A      (2d array):  5 x 4
B      (1d array):      1
Result (2d array):  5 x 4

A      (2d array):  15 x 3 x 5
B      (1d array):  15 x 1 x 5
Result (2d array):  15 x 3 x 5

再来看一些不能进行broadcast的例子:

A  (1d array): 3
B  (1d array): 4        # 最后一维(trailing dimension)不匹配

A  (2d array):      2 x 1
B  (3d array):  8 x 4 x 3(倒数第二维不匹配)

我们再来看一些具体的应用:

>>> x = np.arange(4)
>> xx = x.reshape(4, 1)
>> y = np.ones(5)
>> z = np.ones((3, 4))

>>> x.shape
(4,)
>>> y.shape
(5,)
>>> x+y
ValueError: operands could not be broadcast together with shapes (4,) (5,) 

>>> xx.shape
(4, 1)
>>> y.shape
(5,)
>>> (xx+y).shape
(4, 5)
>>> xx + y
array([[ 1.,  1.,  1.,  1.,  1.],
       [ 2.,  2.,  2.,  2.,  2.],
       [ 3.,  3.,  3.,  3.,  3.],
       [ 4.,  4.,  4.,  4.,  4.]])

当执行xx+y时,numpy是如何进行copy扩展的呢?

xx     (2d array):      4 x 1
y      (1d array):          5
Result (2d array):      4 x 5

也即对xx重复5列,对y重复4行

# 对xx重复5列
# 等价于np.dot(xx, np.ones((1, 4)))
array([[ 0.,  0.,  0.,  0.],
       [ 1.,  1.,  1.,  1.],
       [ 2.,  2.,  2.,  2.],
       [ 3.,  3.,  3.,  3.]])
# 对y重复4行,
array([[ 1.,  1.,  1.,  1.,  1.],
       [ 1.,  1.,  1.,  1.,  1.],
       [ 1.,  1.,  1.,  1.,  1.],
       [ 1.,  1.,  1.,  1.,  1.]])

2. 补充
还有一点,注意和矩阵乘法的区别,当有一维数组参与运算时:
一维数组置于矩阵乘法的左部,被视为一个行向量;
一维数组置于矩阵乘法的右部,被视为一个列向量;
(这样和一个一维数组作用无论在左还是在右)矩阵乘法运算结束得到的向量仍是一维数组。

x   (1d array)  ->  5       ->  1, 5
W   (2d array)  ->  5, 3    ->  5, 3
                            ->  3 (1d array)
>>> import numpy
>>> x = numpy.random.randn(5)
>>> x.shape
(5,)
>>> x.T.shape
(5,)                # 一维数组x并非严格意义上的行向量

>>> W = numpy.random.randn(5, 3)
>>> numpy.dot(x, W)
array([ 0.06489021, -3.08729591,  1.52169767])
>>> numpy.dot(x, W).shape
(3, )

>>> y = numpy.ones(3)
>>> y.shape
(3,)
>>> W.dot(y).shape
(5,)
  • 2
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Python广播机制是指针对两个不同形状的数组进行对应项的加、减、乘、除运算时,首先将数组调整为统一的形状,然后再进行运算。这种机制Numpy、TensorFlow和PyTorch等库都有应用。\[1\] 举个例子来阐述Python广播机制。如果有一个形状为(3,4,5)的三维数组A和一个形状为(4,5)的二维数组B,由于A和B的后缘维度都为(4,5),所以可以进行广播机制。同理,如果A为(3,4)的二维数组,B为(4,)的一维数组,它们的后缘维度都是4,所以也可以进行广播。另外,如果A为(4,5)的三维数组,B为(4,1)的二维数组,两者维度相同,但其一个维度的其一方为1,也可以进行广播。\[2\] 下面是一个验证广播机制的小程序: ```python import numpy as np a = np.array(\[\[1,2,3\],\[4,5,6\]\]) # 2*3 b = np.array(\[\[1\],\[3\]\]) # 2*1 c = a + b print(c) a = np.array(\[\[\[1,2\],\[2,3\],\[3,4\]\],\[\[2,3\],\[4,5\],\[7,8\]\]\]) # 2*3*2 b = np.array(\[\[6,6\],\[7,7\],\[8,8\]\]) # 3*2 c = a + b print(c) print(c.shape) ``` 参考链接:\[https://www.cnblogs.com/jiaxin359/p/9021726.html\](https://www.cnblogs.com/jiaxin359/p/9021726.html) \[2\] 需要注意的是,当两个数组的形状无法满足广播机制的条件时,会抛出ValueError异常。例如,如果数组a的形状为(3,3),数组b的形状为(2,3),那么它们无法进行广播运算,会抛出异常。\[3\] #### 引用[.reference_title] - *1* *3* [【Python学习记录】Numpy广播机制(broadcast)](https://blog.csdn.net/xxm524/article/details/128210631)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [python广播机制broadcasting)](https://blog.csdn.net/weixin_44319196/article/details/107871808)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值