TensforFlow中常用函数总结

TensforFlow中常用函数总结

tf.nn.bias_add(value,bias,name = None)：

一个叫bias的向量加到一个叫value的矩阵上，是向量与矩阵的每一行进行相加，得到的结果和value矩阵大小相同。

import tensorflow as tf
 
a=tf.constant([[1,1],[2,2],[3,3]],dtype=tf.float32)
b=tf.constant([1,-1],dtype=tf.float32)
c=tf.constant([1],dtype=tf.float32)
 
with tf.Session() as sess:
    print('bias_add:')
    print(sess.run(tf.nn.bias_add(a, b)))
    #执行下面语句错误
    #print(sess.run(tf.nn.bias_add(a, c)))

输出：

bias_add:
[[ 2. 0.]
[ 3. 1.]
[ 4. 2.]]

tf.add( x,y, name=None)：

最常见的是，一个叫x的矩阵和一个叫y的数相加，就是y分别与x的每个数相加，得到的结果和x大小相同。

tf.add_n(inputs,name=None)

函数是实现一个列表的元素的相加。就是输入的对象是一个列表，列表里的元素可以是向量，矩阵等，但没有广播功能

tf.squeeze(input,axis=None,name=None,squeeze_dims=None)

该函数返回一个张量，这个张量是将原始input中所有维度为1的那些维都删掉的结果。axis可以用来指定要删掉的为1的维度，此处要注意指定的维度必须确保其是1，否则会报错

tf.expand_dims

tf.expand_dims(
    input,
    axis=None,
    name=None,
    dim=None
)

input是输入张量。
axis是指定扩大输入张量形状的维度索引值。
dim等同于轴，一般不推荐使用。
函数的功能是在给定一个input时，在axis轴处给input增加一个维度。