如何将 3D 阵列的内积转换为二维阵列？

Question

我有两个 Numpy 阵列

b=np.array([[1, 2, 3], [4, 5, 6]])
a=np.array([[[1, 2,1], [3, 4,1],[4,5,6],[6,7,8]], [[5, 6,1], [7, 8,1],[4,5,6],[6,7,8]]])

a.shape,b.shape
((2, 4, 3), (2, 3))

我想计算这些数组的点积。 我尝试了以下代码：

s=np.flip(np.dot(a,b).transpose((0,2,1)),1)

但它会引发以下错误：

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-140-6ada0906cbfc> in <module>
----> 1 s=np.flip(np.dot(a,b).transpose((0,2,1)),1)

<__array_function__ internals> in dot(*args, **kwargs)

ValueError: shapes (2,4,3) and (2,3) not aligned: 3 (dim 2) != 2 (dim 0)

我希望我的结果数组的形状应该是 -(2,4)

我将如何计算这个？ 请建议一些其他方法来做到这一点。

Answer 1

a的形状应该是(*, 3, 2)因为b的形状是(2, 3) 。

a_ · b = │ a11 a12 | * | b11 b12 b13 |
         │ a21 a22 |   | b21 b22 b23 |
         │ a31 a32 | 

其中a_是 a 的a元素。

使用以下a ：

a=np.array([[
    [1, 2], [1, 3], [4, 1]
], [
    [4, 5] ,[6, 6], [7, 8]
], [
    [5, 6], [1, 7], [8, 1]
], [
    [4, 5], [6, 6] ,[7, 8]
]])

我没有错误。 所以试试：

np.dot(a.reshape(4, 3, 2), b)