张量在张量流中沿轴相乘

Question

我使用tensorflow遇到的问题如下：

对于两个张量

[[x11,x12...],[x21,x22...],...[xn1,xn2...]]

和

[y1,y2,...yn],

我想将它们沿轴 0 相乘得到

[[x11*y1,x12*y1...],[x21*y2,x22*y2...]...]

例如，对于 [[1,2],[3,4]]和[1,2]，我想得到结果张量[[1,2],[6,8]]。

实际情况是我有两个张量 A 和 B 形状（批次、高度、宽度、n_channels）和（批次，1）。两者都是在 tensorflow 中定义的张量。对于批次中 A 的每个图像，我想将其与 B 中的相应值相乘。

Answer 1

给定一个二维张量x 和一个向量y，你只需要这样做：

result = x * tf.expand_dims(y, axis=-1)

或者，如果你更喜欢它：

result = x * y[:, tf.newaxis]