Tensorflow: tf.add()

65 阅读 0 评论 43 点赞

我是靠谱客的博主暴躁糖豆，这篇文章主要介绍Tensorflow: tf.add()，现在分享给大家，希望可以做个参考。

Tensorflow: tf.add()

1.基本用法： 单个数字和单个数字的简单相加。

复制代码

1
2
3
4
5
6
7
8
import tensorflow as tf

x = tf.constant(2)
y = tf.constant(1)
z = tf.add(x,y)
with  tf.Session() as sess:
    print(sess.run(z))

输出：3

2.广播机制： 即按维度的相加

在这里插入图片描述
论文《BiSeNet V2: Bilateral Network with Guided Aggregation for Real-time Semantic Segmentation》中的 Context Embedding Block模块，将1x1xC的特征图和HxWxC的特征图进行相加，即采用广播机制的相加。

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
import tensorflow as tf

a = tf.constant([[[2,3]]])
b = tf.constant([[[2,3],
                [1,0,]],[[5,6],[3,2]],[[4,5],[1,8]]])
c = tf.add(a,b)
print(a) #Tensor("Const_2:0", shape=(1, 1, 2), dtype=int32)
print(b) #Tensor("Const_3:0", shape=(3, 2, 2), dtype=int32)
print(c) #Tensor("Add_3:0", shape=(3, 2, 2), dtype=int32)
with tf.Session() as sess:
    print(sess.run(c))