我试图将一个 3D 数组乘以一个 1D 数组,这样沿着第 3 个(深度:d)维度的每个 2D 数组的计算方式如下:
1D_array[d]*2D_array
我最终得到一个看起来像这样的数组,比如:
[[
[1,1]
[1,1]]
[
[2,2]
[2,2]]
[
[3,3]
[3,3]]]
Run Code Online (Sandbox Code Playgroud)
这将是将 np.ones((3,2,2)) 与 [1,2,3] 正确相乘的结果。
我已经尝试了一段时间,无论我似乎做什么,我都无法得到这个结果,只是主题的变化。我该如何正确地执行此操作?
谢谢你的帮助。
python arrays numpy multidimensional-array array-broadcasting