如何乘以不同形状的张量？（广播）

Question

我想对形状为 [a,b,c] 的张量中的元素进行倍增，每个元素在第 3 维中都有一个来自形状为 [a,b] 的张量的标量。

例如，

x = 
|[1,2][3,4][5,6]|
|[1,2][3,4][5,6]|
|[1,2][3,4][5,6]|

y = 
|0 0 1|
|1 0 0|
|0 1 0|

我想乘以 x 和 y 以获得：

res = 
|[0,0][0,0][5,6]|
|[1,2][0,0][0,0]|
|[0,0][3,4][0,0]|

你能告诉我任何方法来实现这一目标吗？

PS：也许我可以做一个y'由y它

y' =
|[0,0][0,0][1,1]|
|[1,1][0,0][0,0]|
|[0,0][1,1][0,0]|

然后我可以tf.mul()用来获得结果。但是我找不到任何以这种方式复制张量中的元素的东西。

Answer 1

你是对的，你可以直接使用tf.mul()。事实上，TensorFlow 中的乘法运算处理广播，因此您不需要创建另一个形状为 的向量[a, b, c]。

为了确保广播在预期的维度上，您可以向第二个向量添加第三个维度y：

x = tf.random_normal([3, 3, 2])
y = tf.constant([[0., 0., 1.], [1., 0., 0.], [0., 1., 0.]])

y = tf.expand_dims(y, 2)  # y will now have a matching shape of [3, 3, 1]
res = tf.mul(x, y)

sess = tf.Session()
sess.run(res)