首页 文章

张量乘以张量流中的轴

提问于
浏览
0

我使用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] ,我想得到结果tensor [[1,2],[6,8]] .

真实情况是我有两个张量A和B形状(批次,高度,宽度,n_channels)和(批次,1) . 两者都是张量流中定义的张量 . 对于批处理中A的每个图像,我想将它与B中的相应值相乘 .

1 回答

  • 0

    给定一个二维张量 x 和一个向量 y ,你只需要做:

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

    或者,如果你更喜欢它:

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

相关问题