标签: numpy elementwise-operations
假设我有N×1×1阵列a,N×M×M阵列b作为NumPy阵列。我想做元素乘法:
a
b
c[i,:,:] = a[i]*b[i,:,:]
没有迭代i。函数np.multiply(a,b)似乎可以完成这项工作。但是,当a和b的大小不同时,我不太了解此函数的内部工作原理。我知道当它具有相同的大小时,它只是元素相乘。我假设当它们的大小不同时它会进行一些广播以改变其中一个阵列的尺寸但是如何?
i
np.multiply(a,b)