python - 将 block 矩阵的每个 block 与 NumPy 中的不同系数相乘

Question

假设我有一个 (4*n)*(4*n) block 矩阵，我想将 (n*n) 的每个 block 乘以与相应的 (4*4) 矩阵不同的系数 - 在 NumPy 中如何做到这一点？

例如，我有:

>>> mat
matrix([[1, 1, 1, 1],
        [1, 1, 1, 1],
        [1, 1, 1, 1],
        [1, 1, 1, 1]])
>>> x
array([[1, 2],
       [3, 4]])

请求的结果应该是这样的:

>>> result
array([[1, 1, 2, 2],
       [1, 1, 2, 2],
       [3, 3, 4, 4],
       [3, 3, 4, 4]])

Answer 1

一种方法是从 x 构造一个 (4*n)*(4*n) block 矩阵:

result = np.multiply(mat, np.kron(x, np.ones((n,n))))