python - Theano:如何获取元素为矩阵的 "matrix outer product"

Question

基本上，我有 2 个张量: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 中高效地完成吗？

旁注:我想要实现的实际最终结果是拥有一个形状为 (N, K, D) 的张量 E，这样:

E[i, j, :] = (A[i, :, :]*B[j, :, :]).sum(0)

因此，如果有办法直接获取它，我会更喜欢它(希望能节省空间)。

Answer 1

可以建议使用 broadcasting 的一种方法-

(A[:,None]*B).sum(2)

请注意，在 axis=2 上的求和减少之前，正在创建的中间数组的形状为 (N, K, H, D) 将其减少为 (N,K,D)。