这与问题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表达式来编写它,有没有办法这样做? 如果有很多,哪一个是最快的?