@rth建议以较小的批次进行操作。您也可以尝试使用该函数
np.subtract并将其分配给目标数组，以避免创建其他临时数组。我也认为您不需要将索引编
c为
c[np.newaxis,:, :]，因为它已经是一个3维数组。
所以代替
a[:] = b[:, :, np.newaxis] - c[np.newaxis, :, :] # memory explodes here
尝试
np.subtract(b[:, :, np.newaxis], c, a)
的第三个参数
np.subtract是目标数组。
