python - Numpy:将矩阵元素与矩阵数组相乘

Question

我需要将 2x2 矩阵 x 的元素与元素为 2x2 矩阵的矩阵 y 相乘。当我使用传统的 numpy 乘法时，它需要整个矩阵 x，并将其与 y 中的每个矩阵相乘。我一直在搜索 numpy 文档。对于将复制此内容的东西:

>>> x = np.array([[1, 0], [0, 1]])
>>> x
array([[1, 0],
       [0, 1]])
>>> y = np.ones((2, 2, 2, 2))
>>> y
array([[[[ 1.,  1.],
         [ 1.,  1.]],
    [[ 1.,  1.],
     [ 1.,  1.]]],
   [[[ 1.,  1.],
     [ 1.,  1.]],
    [[ 1.,  1.],
     [ 1.,  1.]]]])
>>> multiply(x,y)
[[[[1, 1],
   [1, 1]],
  [[0, 0],
   [0, 0]]],
 [[[0, 0],
   [0, 0]],
  [[1, 1],
   [1, 1]]]]

Answer 1

编辑:从@Dalek和@DSM的评论来看，您实际上想要的是:

np.einsum('ij, ijkl-> ijkl', x, y)