小编dxl*_*xli的帖子

Numpy 将 3d 矩阵乘以 2d 矩阵

例如,我得到了形状为 (3,2,2) 的矩阵 A,例如

[
[[1,1],[1,1]], 
[[2,2],[2,2]], 
[[3,3],[3,3]]
]
Run Code Online (Sandbox Code Playgroud)

和形状(2,2)的矩阵B,例如

[[1, 1], [0,1]]
Run Code Online (Sandbox Code Playgroud)

我想实现 c 的形状 (3,2,2) 像:

c = np.zeros((3,2,2))
for i in range(len(A)):
    c[i] = np.dot(B, A[i,:,:])
Run Code Online (Sandbox Code Playgroud)

这使

[[[2. 2.]
  [1. 1.]]

 [[4. 4.]
 [2. 2.]]

 [[6. 6.]
 [3. 3.]]]
Run Code Online (Sandbox Code Playgroud)

实现这一目标的最有效方法是什么?

谢谢。

python numpy

6
推荐指数
1
解决办法
1253
查看次数

标签 统计

numpy ×1

python ×1