簡體   English   中英

NumPy 廣播:在 (M,A,B,N) 形狀數組中跨元素 (a,b) 添加 ndarray

[英]NumPy Broadcast: Adding an ndarray across elements (a,b) in (M,A,B,N) shaped array

我有一個帶有 rgb 像素的圖像數組,因此該數組具有維度(image_count,height,width,3)

我正在通過我在側面計算的隨機數量調整單個像素,形狀為: (image_count,3)

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


我如何廣播這個加法,以便對於每個像素,我執行以下操作:

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

在適當的維度為pixel_adj引入帶有None/np.newaxis新軸使其成為4D ,就像image_set一樣,然后執行求和調整以引入broadcasting ,就像這樣 -

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

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM