Tensordot混乱

Question

我可能误会了tensordor。我正在尝试进行以下部分收缩：

c(e,q,i,j) = a(e,q,i,j,k,l) * b(e,q,l,k)

使用

import numpy as np

a = np.random.random(1*4*2*2*2*2).reshape(1,4,2,2,2,2)
b = np.random.random(1*4*2*2).reshape(1,4,2,2)

c = np.tensordot(a,b,axes=([5,2],[4,3]))

但这给了我错误

/usr/local/lib/python3.7/site-packages/numpy/core/numeric.py in tensordot(a, b, axes)
   1282     else:
   1283         for k in range(na):
-> 1284             if as_[axes_a[k]] != bs[axes_b[k]]:
   1285                 equal = False
   1286                 break

IndexError: tuple index out of range

我误会什么？

Answer 1

我们希望保持很少的轴对齐。因此，tensordot不能直接工作。相反，我们可以使用np.einsum-

np.einsum('ijklmn,ijnm->ijkl',a,b)