利用Numpy數(shù)組如何實(shí)現(xiàn)廣播機(jī)制

發(fā)布時(shí)間：2020-11-04 14:29:14 來源：億速云閱讀：248 作者：Leah 欄目：開發(fā)技術(shù)

這篇文章運(yùn)用簡(jiǎn)單易懂的例子給大家介紹利用Numpy數(shù)組如何實(shí)現(xiàn)廣播機(jī)制，內(nèi)容非常詳細(xì)，感興趣的小伙伴們可以參考借鑒，希望對(duì)大家能有所幫助。

數(shù)組廣播

數(shù)組在進(jìn)行矢量化運(yùn)算的時(shí)，要求數(shù)組形狀時(shí)相等的。當(dāng)形狀不等的數(shù)組執(zhí)行算術(shù)運(yùn)算的時(shí)候，就會(huì)出現(xiàn)廣播機(jī)制，該機(jī)制會(huì)對(duì)數(shù)組進(jìn)行擴(kuò)展，使數(shù)組的shape屬性值一樣，就可以進(jìn)行矢量化運(yùn)算了。

import numpy as np
arr1 = np.array([[0], [1], [2], [3]])
print(arr1.shape)
arr2 = np.array([1, 2, 3])
print(arr2.shape)
val = arr2+arr1
print(val)

輸出結(jié)果
(4, 1) //arr1的shape
(3,) //arr2的shape
[[1 2 3]
[2 3 4]
[3 4 5]
[4 5 6]]

由于arr1和arr2的shape不等，故通過廣播機(jī)制進(jìn)行如下圖的擴(kuò)展

利用Numpy數(shù)組如何實(shí)現(xiàn)廣播機(jī)制

廣播機(jī)制的使用條件

1、數(shù)組的某一維度等長(zhǎng)
2、其中一個(gè)數(shù)組的某一維度為1

關(guān)于利用Numpy數(shù)組如何實(shí)現(xiàn)廣播機(jī)制就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，可以學(xué)到更多知識(shí)。如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到。

向AI問一下細(xì)節(jié)

利用Numpy數(shù)組如何實(shí)現(xiàn)廣播機(jī)制

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽