使用 Tensorflow 的批量矩阵乘法

Question

Tensor A is [M X 2N X N]
Tensor B is [M X N]

我需要将两个张量相乘得到：

Tensor C [M X 2N X N]. 

这是一个例子：

M= 2, N = 2

A: [[[1,2]
     [1,2]
     [1,2]
     [1,2] ]

    [[2,2]
     [2,2]
     [2,2]
     [2,2] ]]

 B = [[3,3]
      [2,2]]

 C: [[[3,6]
      [3,6]
      [3,6]
      [3,6] ]

      [[4,4]
       [4,4]
       [4,4]
       [4,4] ]]

不知道如何实现这一点。除了解决方案，有人可以解释一下广播在这种情况下是如何工作的吗？

Answer 1

看起来你想要元素乘法，其中张量 B 在该维度上重复 2N 次。我会使用 tf.concat(2, [tf.reshape(b, [m, 1, n]) for _ in xrange(2*n)]) 手动进行广播。