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