NumPy的廣播：橫跨元件（A，B）在（M，A，B，N）型陣列

Question

我與RGB像素圖像的陣列，以使陣列具有的尺寸(image_count,height,width,3)

我添加一個ndarray用形狀爲(image_count,3)的一側計算的隨機量I調整各個像素。

現在，我有一個解釋循環，將調整添加到單個像素。

---

我怎麼播出這個此外，這樣，對於每個像素，我做以下操作：

image_set[i,h,w,:] += pixel_adj[i,:] 

Answer 1

在合適的尺寸引進新的軸與None/np.newaxis爲pixel_adj，使其4D，就像image_set，然後執行總和調整，以引入broadcasting，像這樣 -

image_set += pixel_adj[:,None,None,:]