gpt4 book ai didi

python - 使用 theano 的矩阵三重积

转载 作者:行者123 更新时间:2023-11-28 16:29:58 24 4
gpt4 key购买 nike

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

所以我有三个矩阵ABC,大小为n*r m*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 表达式来写它,有办法吗?如果有很多,哪个最快?

最佳答案

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)

关于python - 使用 theano 的矩阵三重积,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32966627/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com