1

假设我有一个 N × 1 × 1 数组a和 N × M × M 数组b作为 NumPy 数组。我想做元素乘法:

c[i,:,:] = a[i]*b[i,:,:]

无需迭代i。该功能np.multiply(a,b)似乎可以完成这项工作。但是,我不太了解这个函数的内部工作原理,a并且b没有相同的大小。我知道当它具有相同的大小时,它只会按元素相乘。我假设当它们的大小不同时,它会进行一些广播以更改其中一个数组的尺寸,但是如何?

4

0 回答 0