小编use*_*581的帖子

numpy 多维(3d)矩阵乘法

我得到两个 3d 矩阵 A(32x3x3)和 B(32x3x3),我想得到维度为 32x3x3 的矩阵 C。计算可以使用循环完成,如:

a = numpy.random.rand(32, 3, 3)
b = numpy.random.rand(32, 3, 3)
c = numpy.random.rand(32, 3, 3)

for i in range(32):
    c[i] = numpy.dot(a[i], b[i])
Run Code Online (Sandbox Code Playgroud)

我相信必须有一个更有效的单线解决方案来解决这个问题。有没有人可以帮忙,谢谢。

python numpy matrix multidimensional-array matrix-multiplication

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