前言

在吳恩達(dá)老師的深度學(xué)習(xí)專項(xiàng)課程中，老師有提到NumPy中的廣播機(jī)制，同時(shí)那一周的測(cè)驗(yàn)也有涉及到廣播機(jī)制的題目。那么，到底什么是NumPy中的廣播機(jī)制？

官方文檔

接下來(lái)到了看官方文檔的時(shí)間。

Array Broadcasting in Numpy

廣播機(jī)制概述

讓我們探索numpy中一個(gè)更高級(jí)的概念，這個(gè)概念被稱為廣播。 廣播展現(xiàn)了NumPy在算術(shù)運(yùn)算期間是如何處理具有不同形狀的數(shù)組的。 受到某些約束，較小的陣列將在較大的陣列上“廣播”，以使它們具有相同形狀。 廣播提供了一種數(shù)組矢量化操作，從而使得循環(huán)在C而不是Python中發(fā)生。 它無(wú)需復(fù)制不必要的數(shù)據(jù)即可完成，并且通常算法的效率還挺高。 當(dāng)然在某些情況下，廣播并不是一個(gè)好辦法，因?yàn)樗鼤?huì)導(dǎo)致內(nèi)存使用效率低，從而減慢計(jì)算速度。 本文通過(guò)示例，對(duì)廣播進(jìn)行了詳盡的介紹。 它還提供何時(shí)使用廣播的提示。

numpy操作通常是逐個(gè)元素完成的，這就需要兩個(gè)數(shù)組具有完全相同的形狀

Example 1

>>> from numpy import array
>>> a = array([1.0, 2.0, 3.0])
>>> b = array([2.0, 2.0, 2.0])
>>> a * b
array([ 2.,  4.,  6.])

當(dāng)數(shù)組的形狀滿足某些條件時(shí)，numpy的廣播規(guī)則將放寬這種數(shù)組限制。 將數(shù)組和標(biāo)量值在一起運(yùn)算時(shí)，會(huì)出現(xiàn)最簡(jiǎn)單的廣播示例：

Example 2

>>> from numpy import array
>>> a = array([1.0,2.0,3.0])
>>> b = 2.0
>>> a * b
array([ 2.,  4.,  6.])

盡管只有一個(gè)變量是數(shù)組，但是結(jié)果和之前的一個(gè)代碼例子是一樣的。 我們可以認(rèn)為其中的標(biāo)量在算術(shù)運(yùn)算中被拓展成與數(shù)組a變量形狀相同的數(shù)組。 例如下圖中顯示的中拓展的新元素只是原始標(biāo)量的副本。這種拓展只是概念上的。 numpy的明智之處在于使用原始標(biāo)量值而不必要?jiǎng)?chuàng)建副本，從而使廣播操作盡可能地節(jié)省內(nèi)存提高計(jì)算效率。 由于上面的代碼例子中，乘法過(guò)程中標(biāo)量移動(dòng)的內(nèi)存較少，所以在具有一百萬(wàn)個(gè)元素?cái)?shù)組的Windows 2000上，廣播機(jī)制與之前的兩個(gè)數(shù)組相加相比大概快10%。

在最簡(jiǎn)單的廣播示例中，標(biāo)量b被拉伸為與a相同形狀的數(shù)組，使得這些形狀適用于逐元素乘法。

下面的規(guī)則決定了兩個(gè)具有兼容形狀的數(shù)組是否可以在單個(gè)代碼段中進(jìn)行廣播。

廣播機(jī)制規(guī)則

廣播規(guī)則

為了廣播，操作中兩個(gè)陣列的尾軸的大小必須相同，或者其中一個(gè)必須是一個(gè)。

問(wèn)題來(lái)了，尾軸是什么？

為此我找到了python - numpy broadcasting - explanation of trailing axes - Stack Overflow這篇解答。

If you have two arrays with different dimensions number, say one 1x2x3 and other 2x3, then you compare only the trailing common dimensions, in this case 2x3. But if both your arrays are two-dimensional, then their corresponding sizes have to be either equal or one of them has to be 1.

In your case you have a 2x2 and 4x2 and 4 != 2 and neither 4 or 2 equals 1, so this doesn't work.

假設(shè)你有兩個(gè)不同維度的數(shù)組。一個(gè)是1x2x3，另一個(gè)是2x3，那么只需要比較后面的公共尺寸，在這種情況下為2x3。 但是，**如果兩個(gè)數(shù)組都是二維的，則它們的對(duì)應(yīng)大小必須相等或其中之一必須為1 **。

在兩個(gè)二維數(shù)組中2x2和4x2，4！= 2，并且4或2都不等于1，所以廣播行不通的。

這個(gè)解釋應(yīng)該比較清楚了。

如果不滿足此條件，則會(huì)引發(fā)異常，提示數(shù)組的形狀不兼容。 廣播操作創(chuàng)建的結(jié)果數(shù)組的大小是兩個(gè)數(shù)組中每個(gè)維度的最大大小。 請(qǐng)注意，該規(guī)則并未說(shuō)明需要具有相同維數(shù)的兩個(gè)數(shù)組。 如果有一個(gè)256 x 256 x 3的RGB值數(shù)組，想要按不同的值縮放圖像中的每種顏色，則可以將圖像乘以具有3個(gè)值的一維數(shù)組。