我有一个形状(10, 3)
的2D
数组和一个表示为形状(480, 640, 3)
的3D
数组的图像.我想对2D
数组中的每个像素和每个元素进行差值运算,以得到形状(10, 480, 640, 3)
的最终结果.
目前,我的代码如下所示:
arr_2d = np.random.rand(10, 3)
arr_3d = np.random.rand(480, 640, 3)
res = np.ones_like(arr_3d)
res = np.tile(res, (10, 1, 1, 1))
for i in range(10):
res[i] = arr_3d - arr_2d[i]
我的问题是,是否有一种方法可以在不使用for循环的情况下做到这一点,只使用NumPy操作.