如何解决Python：for循环内的einsum
假设 A 和 B 是两个具有相同维度的 4 维 numpy 数组。
A = np.random.rand(5,5,2,10)
B = np.random.rand(5,10)
a,b,c,d = A.shape
dat = []
for k in range(d):
sum = 0
for l in range(c):
sum = sum + np.einsum('ij,ji->',A[:,:,l,k],B[:,k])
dat.append(sum)
我想知道是否可以使用“einsum”来替换内部 for 循环，甚至是外部 for 循环，或者一些矩阵操作来替换所有循环，因为数据集很大。
有没有更快的方法来实现这一目标？
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
