张量tensorflow

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]