python-带theano的矩阵三元产品

这个问题和这里的Matrix/Tensor Triple Product?差不多,只是theano的问题.

所以我有三个矩阵A,B,C,它们的大小分别为n * r,m * r,l * r,并且我想计算由三重(三线性)乘积得到的形状为(n,m,l)的3D张量：

X[i,j,k] = \sum_a A[i,a] B[j,a] C[k,a]

A,B和C是共享变量：

A = theano.shared(numpy.random.randn(n,r))
B = theano.shared(numpy.random.randn(m,r))
C = theano.shared(numpy.random.randn(l,r))

我想用一个theano表达式编写它,有办法吗？
如果有很多,哪一个是最快的？

解决方法:

np.einsum('nr,mr,lr->nml', A, B, C)

相当于

np.dot(A[:, None, :] * B[None, :, :], C.T)

可以在Theano中实现为

theano.dot(A[:, None, :] * B[None, :, :], C.T)

标签：blas,linear-algebra,matrix-multiplication,theano,python
来源： https://codeday.me/bug/20191119/2037631.html