矩阵三重产品与theano

时间:2015-10-06 09:38:41

标签: python linear-algebra matrix-multiplication theano blas

这与问题Matrix/Tensor Triple Product?几乎相同,但对于theano。

所以我有三个矩阵ABC大小n*rm*rl*r,我想要计算由三重(三线性)乘积产生的形状(n,m,l)的3D张量:

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

ABC是共享变量:

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

0 个答案:

没有答案