python - 将 einsum 计算转换为要在 Theano 中使用的点积

Question

我最近才了解 np.einsum 并很快就对它上瘾了。但似乎 theano 没有等效的功能，所以我需要以某种方式将我的 numpy 代码转换为 theano。如何在 theano 中编写以下计算？

IX=np.einsum('ijk,lj->ilk',p1,KX)

Answer 1

你只需要重新排列你的坐标轴就可以让它工作:

>>> import numpy as np
>>> a = np.random.rand(3, 4, 5)
>>> b = np.random.rand(5, 6)
>>> np.allclose(np.einsum('ikj,jl->ikl', a, b), np.dot(a, b))

考虑到这一点:

>>> a = np.random.rand(3, 5, 4)
>>> b = np.random.rand(6, 5)
>>> out_ein = np.einsum('ijk,lj->ilk', a, b)
>>> out_dot = np.transpose(np.dot(np.transpose(a, (0, 2, 1)),
...                               np.transpose(b, (1, 0))),
...                        (0, 2, 1))
>>> np.allclose(out_ein, out_dot)