Tensorflow常用函数说明

Posted Qniguoym

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Tensorflow常用函数说明相关的知识,希望对你有一定的参考价值。

1.矩阵操作

1.1矩阵生成

这部分主要将如何生成矩阵,包括全0矩阵,全1矩阵,随机数矩阵,常数矩阵等

sess=tf.InteractiveSession()
#x=tf.ones([2,3],tf.int32)
x=tf.zeros([2,3],tf.int32)
print (sess.run(x))

新建一个与给定的tensor类型大小一致的tensor,使其所有元素为0和1

sess=tf.InteractiveSession()
tensor=[[1,2,3],[4,5,6]]
#x=tf.ones_like(tensor)
x=tf.zeros_like(tensor)
print (sess.run(x))

创建一个形状大小为shape的tensor,初始值为value

sess=tf.InteractiveSession()
x=tf.fill([2,3],2)
print (sess.run(x))

创建一个常量tensor,按照给出value来赋值,可以用shape来指定其形状。value可以是一个数,也可以是一个list。 
如果是一个数,那么这个常量中所有值的按该数来赋值。 
如果是list,那么len(value)一定要小于等于shape展开后的长度。赋值时,先将value中的值逐个存入。不够的部分,则全部存入value的最后一个值。

sess=tf.InteractiveSession()
#x=tf.constant(2,shape=[2,3])
x=tf.constant([2,3],shape=[2,3])
print (sess.run(x))

random_normal:正态分布随机数,均值是mean,标准差是stddev

truncated_normal:截断正态分布随机数,均值是mean,标准差是stddev,只保留[mean-2*stddev,mean+2*stddev]范围内的随机数

random_uniform:均匀分布随机数,范围为[minval,maxval]

sess=tf.InteractiveSession()
x=tf.random_normal(shape=[1,5],mean=0.0,stddev=1.0,dtype=tf.float32)
print (sess.run(x))
tf.get_variable(name, shape=None, dtype=dtypes.float32, initializer=None,
                 regularizer=None, trainable=True, collections=None,
                 caching_device=None, partitioner=None, validate_shape=True,
                 custom_getter=None):

如果在该命名域中之前已经有名字=name的变量,则调用那个变量;如果没有,则根据输入的参数重新创建一个名字为name的变量。

initializer为初始化工具,有tf.zero_initializer,tf.ones_initializer,tf.constant_initializer,tf.random_uniform_initializer,tf.random_normal_initializer,tf.truncated_normal_initializer

1.2 矩阵变换

tf.shape(Tensor)

返回张量的形状。但是注意,tf.shape函数本身也是返回一个张量。而在tf中,张量是需要用sess.run(Tensor)来得到具体的值的。

sess=tf.InteractiveSession()
x=[[1,2,3],[4,5,6]]
shape=tf.shape(x)
print (shape)
print (sess.run(shape))

Tensor("Shape:0", shape=(2,), dtype=int32)
[2 3]

tf.expand_dims(Tensor,dim)

为张量增加一个维度

sess=tf.InteractiveSession()
x=[1,2,3]
y=tf.expand_dims(x,1)
print (sess.run(y))

[[1]
 [2]
 [3]]

#y=tf.expand_dims(x,0)
[[1,2,3]]

tf.pack()

将一个R维张量列表沿着axis轴组合成一个R+1维的张量。

x=[1,4]
y=[2,5]
z=[3,6]
t=tf.stack([x,y,z],1)
print (sess.run(t))

[[1 2 3]
 [4 5 6]]

tf.concat(concat_dim,values)将张量沿着指定的维数拼接起来。

tf.sparse_to_dense()将稀疏矩阵转化为密集矩阵

def sparse_to_dense(sparse_indices,
                    output_shape,
                    sparse_values,
                    default_value=0,
                    validate_indices=True,
                    name=None):

sparse_indices: 元素的坐标[[0,0],[1,2]] 表示(0,0),和(1,2)处有值 
output_shape: 得到的密集矩阵的shape 
sparse_values: sparse_indices坐标表示的点的值,可以是0D或者1D张量。若0D,则所有稀疏值都一样。若是1D,则len(sparse_values)应该等于len(sparse_indices) 
default_values: 缺省点的默认值

tf.random_shuffle(value)

沿着value的第一维进行随机重新排列。

tf.argmax|tf.argmin(input=tensor,dimension=axis)

找到给定的张量tensor在指定轴axis上的最大值/最小值的位置。

tf.equal(x,y)

判断两个tensor是否每个元素都相等。

tf.cast(x,dtype)

将x的数据格式转化为dtype。

tf.matmul(a,b,transpose_a=False,transpose_b=False,a_is_parse=False,b_is_sparse=False)

如果对应的transpose项为True,例如transpose_a=True,那么a在参与运算之前就会先转置一下。 
而如果a_is_sparse=True,那么a会被当做稀疏矩阵来参与运算。

tf.reshape(tensor,shape)

如果shape=[-1],表示将tensor展开成一个list。

2. 神经网络相关操作

tf.nn.embedding_lookup(params,ids)

简单来说,就是讲一个数字序列ids转化成embedding序列表示。

tf.trainable_variables()

返回所有可以训练的变量。

tf.gradients(ys,xs)

ys和xs都可以是一个tensor或者tensor列表。而计算完成以后,该函数会返回一个长为len(xs)的tensor列表,列表中的每个tensor是ys中每个值对xs[i]求导之和。

tf.clip_by_global_norm(t_list,clip_norm)

修正梯度值,用于控制梯度爆炸的问题。梯度爆炸和梯度弥散的原因一样,都是因为链式法则求导的关系,导致梯度的指数级衰减。为了避免梯度爆炸,需要对梯度进行修剪。

t_list为待修剪的张量,clip_norm表示修剪比例,函数返回两个参数:list_clipped,修剪后的张量,以及global_norm,一个中间计算量。

list_clipped[i]=t_list[i] * clip_norm / max(global_norm, clip_norm),其中 
global_norm = sqrt(sum([l2norm(t)**2 for t in t_list]))

tf.nn.dropout(x,keep_prob)

按照概率将x中的一些元素置0,并将其他的值放大。x是一个张量,而keep_prob是一个(0,1]之间的值。x中的各个元素清零的概率互相独立,为1-keep_prob,而没有清零的元素,则会统一乘以1/keep_prob, 目的是为了保持x的整体期望值不变。

3.普通操作

tf.linspace|tf.range

tf.linspace(start,stop,num)

在[start,stop]的范围内产生num个数的等差数列,start和stop要用浮点数进行表示。

tf.range(start,limit=None,delta=1) 在[start,limit)范围内以步进值delta产生等差数列。

tf.assign(ref,value)

用来更新模型中变量的值,ref是待赋值的变量,value是更新的值。

4.规范化

tf.variable_scope

为变量添加命名域

with tf.variable_scope("foo"):

tf.get_variable_scope()返回当前变量的命名域。

 

以上是关于Tensorflow常用函数说明的主要内容,如果未能解决你的问题,请参考以下文章

C#常用代码片段备忘

Tensorflow一些常用基本概念与函数

AI常用框架和工具丨9. 深度学习框架TensorFlow

0.tensorflow——常用说明

21个常用代码片段

常用Javascript代码片段集锦