[英]In Numpy, how can I index an B x N x M array with a size B array of indices to yield a B x M array?
[英]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.