我有两个numpy数组x和y,形状分别为(m,n,p)和m。对于0和m-1之间的所有索引i,我想将块x[i,:,:]中的每个元素乘以y[i]。
x
y
(m,n,p)
m
0
m-1
i
x[i,:,:]
y[i]
我如何在一行中实现这一点?
相关问题:How to multiply numpy 1D with N-D array?
jm81lzqq1#
可能的解决方案:
x * y[:, None, None]
字符串
1条答案
按热度按时间jm81lzqq1#
可能的解决方案:
字符串