张量沿轴乘以十

Question

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

对于两个张量

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

以及

^{pr2}$

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

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

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

真实的场景是，我有两个张量A和B形（批次、高度、宽度、n_通道）和（批次，1）。两者都是张量流中定义的张量。对于批处理中A的每个图像，我想将其与B中的相应值相乘

Answer 1

给定一个二维张量x和一个向量y，您只需要：

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

或者，如果你更喜欢：

^{pr2}$