陣列的廣播機制

2021-10-01 02:16:19 字數 694 閱讀 1137

1、讓所有輸入陣列都向其中shape最長的陣列看齊,shape中不足的部分都通過在前面加1補齊。

2、輸出陣列的shape是輸入陣列shape的各個軸上的最大值。

3、如果各個輸入陣列的對應軸的長度相同或者其長度為1時,這樣的陣列之間能夠用來計算,否則

出錯。4、 當輸入陣列的某個軸的長度為1時,沿著此軸運算時都用此軸上的第一組值

如下面**所示,可以滿足廣播的原則,可以進行陣列的加法運算

#建立二維陣列

arr1 = np.arange(4).reshape(2,2)

print(arr1)

print(arr1.shape) #(2, 2)

'''[[0 1]

[2 3]]

(2, 2)'''

arr2 = np.array([1,2])

print(arr2)

print(arr2.shape) #(2,)

'''[1 2]

(2,)'''

廣播原則的例子如下:

Numpy陣列的廣播機制

numpy陣列不需要迴圈遍歷,即可對每個元素執行批量的算術運算操作 向量化運算 當兩個陣列大小 numpy.shape 不同時,進行算術運算會出現廣播機制。陣列在進行向量化運算的時,要求陣列形狀時相等的。當形狀不等的陣列執行算術運算的時候,就會出現廣播機制,該機制會對陣列進行擴充套件,使陣列的sha...

Numpy中陣列的廣播機制總結

numpy中的對於陣列間的算術運算採用 元素一 一對應 的計算機制,因而一般要求兩個陣列的形狀相同才能進行陣列間的算術運算,但是在某些情況中,numpy中允許符合一定規則的不同形狀的陣列進行算術運算。廣播的核心規則如下 上面的規則可能看起來比較難懂,簡單來說就是低維陣列要想被拓展成和高維陣列一樣的形...

pytorch的廣播機制

廣播機制,就是將不同維度 不同長度的tensor,在滿足一定規則的前提下能夠自動進行長度和維度的擴充,從而使不同維度 不同長度的tensor之間正確的進行運算。自動廣播規則 兩個tensor能夠進行自動廣播需要滿足以下幾個規則 對應相等 其中乙個tensor的大小等於1 其中乙個tensor的某個維...