缩放 3D 张量行

Question

我有一个n-by-3-by-3 numpy 数组A 和一个n-by-3 numpy 数组B。我现在想将n3-by-3 矩阵的每一行的每一行与B 中的相应标量相乘，即，

import numpy as np

A = np.random.rand(10, 3, 3)
B = np.random.rand(10, 3)

for a, b in zip(A, B):
    a = (a.T * b).T
    print(a)

没有循环也可以做到这一点吗？

Answer 1

您可以使用NumPy broadcasting 在将B 扩展到3D 之后，在使用np.newaxis 或其别名/简写None 的末尾添加一个单一维度后，让元素乘法以矢量化方式发生。因此，实现将是A*B[:,:,None] 或简单的A*B[...,None]。