張量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]