我有兴趣计算一个大的NumPy数组.我有一个A包含大量数字的大型数组.我想计算这些数字的不同组合的总和.数据结构如下:
A = np.random.uniform(0,1, (3743, 1388, 3))
Combinations = np.random.randint(0,3, (306,3))
Final_Product = np.array([ np.sum( A*cb, axis=2) for cb in Combinations])
我的问题是,如果有更优雅和记忆效率更高的计算方法吗？np.dot()当涉及三维阵列时,我觉得这很令人沮丧.
如果它有帮助,理想的形状Final_Product应该是(3743,306,1388).目前Final_Product形状(306,3743,1388),所以我可以重塑到那里.