numpy - 张量点混乱

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

我有什么误解？

标签： numpy

Answer 1

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

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