我得到两个 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