本文介绍了多维3维阵列乘以2维阵列的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我有两个矩阵，m0.shape = [10, 3, 3]和m1.shape = [10, 3]。我想要做的是使用循环这样做：
m0 = np.zeros((10, 3, 3))
m1 = np.zeros((10, 3))
a = np.zeros((10, 3))
for i in range(10):
a += m1 @ m0[i]
问题是：我可以通过使用内置Numy操作以某种方式达到相同的结果吗？
推荐答案
我认为您有两个选择：
import numpy as np
np.sum(m1 @ m0, axis=0)
或使用numpy.einsum
np.einsum('ij,kjl->il', m1, m0)
这篇关于多维3维阵列乘以2维阵列的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！