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）。两者都是张量流中定义的张量。对于批次中A的每个图像，我想将其与B中的相应值相乘。

Answer 1

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

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

或者，如果您更喜欢它：

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