我有一个维度为(N, M)的数组x和一个维度为(N, M, 2)的数组y。对于任何i, j,我想将每个y[i, j, :]乘以x[i, j]。我该怎么做?
(N, M)
x
(N, M, 2)
y
i, j
y[i, j, :]
x[i, j]
import numpy as np x = np.random.randn(100, 10) y = np.random.randn(100, 10, 2) x * y # this does not work
mfuanj7w1#
按如下方式添加额外的暗淡:
import numpy as np x = np.random.randn(100, 10) print(x[..., None].shape) y = np.random.randn(100, 10, 2) x[..., None] * y
1条答案
按热度按时间mfuanj7w1#
按如下方式添加额外的暗淡: