[经验日记]tensorflow的一种广播机制

最新推荐文章于 2021-11-24 10:01:42 发布

ZAX130

最新推荐文章于 2021-11-24 10:01:42 发布

阅读量176

点赞数

分类专栏： python keras 文章标签： tensorflow 深度学习神经网络

本文链接：https://blog.csdn.net/ZAX130/article/details/115691147

版权

4 篇文章 0 订阅

订阅专栏

3 篇文章 0 订阅

订阅专栏

问题:将shape=(2,)维的Tensor A和shape=(2,2,2)的Tensor B以某种方式相乘,这里它们的第一个维度2代表Batch_size, A可以理解为sigmoid输出, B为两张2x2的图像分割结果,要将A中的数分别对应乘上B中的图片(先验二分类)

A = tf.constant([1,2])
B = tf.constant([
    [
        [1,0],
        [0,1]
    ],
    
    [
        [1,1],
        [0,0]
    ]
])

A*B
直接A*B的结果是这样的,可以发现其实是A与B的后两维做乘法,不满足我们的需求

A = tf.expand_dims(A,axis=-1)
A = tf.expand_dims(A,axis=-1)

扩增维度后A*B
这样在A的后面扩增两维后,A的shape为(2,1,1), 这样A与B维度相同,且第一个维度都是2, A的后两维便广播成2x2与B的后两维匹配,这样就能让对应的数乘到对应的图片上。
（只扩增一次的话还是与第一次结果一样，只是最后两维相乘）

关注