theano：兩個矩陣之間的按行外積

Question

我試圖不使用掃描來計算theano中兩個矩陣之間的按行外積。 我可以使用einsum在numpy中完成此操作，而theano則不可用。

A = np.array([[1,1,1],[2,2,2]])
B = np.array([[3,3,3,3],[4,4,4,4]])
print np.einsum('xi,xj->xij', A, B)
[[[3 3 3 3]
  [3 3 3 3]
  [3 3 3 3]]

 [[8 8 8 8]
  [8 8 8 8]
  [8 8 8 8]]]

Answer 1

這可以通過重塑來實現：許多簡單的einsum操作可以歸結為這一點。 復雜的沒有。

import theano
import theano.tensor as T
import numpy as np

a = np.array([[1,1,1],[2,2,2]]).astype('float32')
b = np.array([[3,3,3,3],[4,4,4,4]]).astype('float32')

A = T.fmatrix()
B = T.fmatrix()

C = A[:, :, np.newaxis] * B[:, np.newaxis, :]

print C.eval({A:a, B:b})

結果是

[[[ 3.  3.  3., 3.]
  [ 3.  3.  3., 3.]
  [ 3.  3.  3.. 3.]]

 [[ 8.  8.  8., 8.]
  [ 8.  8.  8., 8.]
  [ 8.  8.  8., 8.]]]