例如,我得到了形状为 (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)
实现这一目标的最有效方法是什么?
谢谢。