我有两个要乘的数组。第一个尺寸为1000x2x1，另一个尺寸为1000x1x2。我希望结果的尺寸为1000x2x2。通过在阵列上循环1000次可以很容易地做到这一点，但是效率很低。
如何在没有for循环的情况下完成此任务？ 谢谢
答案 0 :(得分：0)
这可以通过在两个数组之间使用*运算符来实现。有关更多信息，请参见numpy broadcasting
arr1 = np.random.rand(1000, 2, 1)
arr2 = np.random.rand(1000, 1, 2)
arr3 = arr1 * arr2
print(arr3.shape) #(1000, 2, 2)