tensorflow知识点

1.tf.reduce_sum

官方的api:

reduce_sum(
    input_tensor,
    axis=None,
    keep_dims=False,
    name=None,
    reduction_indices=None)

input_tensor:表示输入 
axis:表示在那个维度进行sum操作。 
keep_dims:表示是否保留原始数据的维度,False相当于执行完后原始数据就会少一个维度。 

例子:

a = np.ones(24).reshape(2,3,4)
a_sum1 = tf.reduce_sum(
    a,
    axis=[0,1,2]
    )
a_sum2 = tf.reduce_sum(
    a,
    axis=[0]
    )
with tf.Session() as sess:
    print(sess.run(a_sum1))
    print(sess.run(a_sum2))


输出结果为:
24.0
[[2. 2. 2. 2.]
 [2. 2. 2. 2.]
 [2. 2. 2. 2.]]

2.tf.abs(Tensor)

对输入数据(Tensor)每一个元素取绝对值

3.tf.less(Tensor, thresh)

对于tensor中小于thresh的元素,返回True,否则返回False,例子:

a = np.random.random(6).reshape(2,3)
a_less = tf.less(a,0.1)

with tf.Session() as sess:
    print(sess.run(a_less))


输出:
[[False  True False]
 [False False False]]

猜你喜欢

转载自blog.csdn.net/Mr_health/article/details/84844988