基本上,我有两个张量:A,其中A.shape = (N, H, D),和B,其中B.shape = (K, H, D).我想做的是得到一个张量C,其形状(N, K, D, H)如下:
C[i, j, :, :] = A[i, :, :] * B[j, :, :].
这可以在Theano中有效地完成吗？
附注:我想要实现的实际最终结果是具有形状的张量E,(N, K, D)使得:
E[i, j, :] = (A[i, :, :]*B[j, :, :]).sum(0)
所以,如果有办法直接得到这个,我更喜欢它(希望节省空间).