缩放3D张量行

Question

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

B

这可以在没有循环的情况下完成吗？

Answer 1

在B或其末尾添加单个维度后，您可以使用NumPy broadcasting将3D扩展为np.newaxis后，以矢量化方式进行元素乘法运算别名/速记None。因此，实施将是A*B[:,:,None]或仅A*B[...,None]。