我有一个包含 RGB 像素的图像数组，因此该数组的尺寸为 (image_count,height,width,3)
我正在按照我在侧面计算的随机量调整各个像素，形状为:(image_count,3)。
现在，我有一个解释循环来将调整添加到各个像素。
如何广播此加法，以便对每个像素执行以下操作:
image_set[i,h,w,:] += pixel_adj[i,:]
最佳答案
通过 None/np.newaxis 引入新轴以适当的尺寸进行 pixel_adj 使其成为 4D，就像 image_set 一样，然后执行求和调整以引入 broadcasting ，就像这样 -
image_set += pixel_adj[:,None,None,:]
关于python - NumPy 在 (M,A,B,N) 形状数组中广播 : Adding an ndarray across elements (a, b)，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/36825588/