tensorflow常见函数
1、类型转换
tf.string_to_number(string_tensor,out_type=None,name=None) #字符串转为数字
tf.to_double(x,name='ToDouble') #转为64位浮点类型
tf.to_float(x,name='ToFloat') #转为32位浮点类型
tf.to_int32(x,name='ToInt32') #转为32位整型
tf.to_int64(x,name='ToInt64') #转为64位整型
tf.cast(x,dtype,name=None) #将x或者x.values转换为dtype所指定的类型
2、数值操作
tf.ones(shape,stype) #生成1的张量。tf.ones([2,3],tf.int32)
tf.zeros(shape,dtype) #生成0的张量。
tf.ones_like(input) #生成和输入张量一样形状和类型的1。
tf.zeros_like(input) #生成和输入张量一样形状和类型的0。
tf.fill(shape,value) #为指定形状填值。
tf.constant(value,shape) #生成常量。
tf.random_normal(shape, mean=0.0, stddev=1.0, dtype=tf.float32, seed=None, name=None) #正太分布随机数
tf.truncated_normal(shape, mean=0.0, stddev=1.0, dtype=tf.float32, seed=None, name=None) #截断正太分布随机数
tf.random_uniform(shape, minval=0, maxval=None, dtype=tf.float32, seed=None, name=None) #均匀分布随机数
tf.random_crop(value, size, seed=None, name=None) #将输入值value按照size尺寸随机剪辑
tf.set_random_seed(seed) #设置随机数种子
tf.linspace(start, stop, num, name=None) #在[start,stop]范围内产生num个数的等差数列。start,stop要用浮点数表示。
tf.range(start,limit=None,delta=1,name='range') #在[start,limit)范围内以步进值delta产生等差数列。
3、形状变换
tf.shape(input,name=None) #返回一个张量，其值为输入参数input的shape。这个input可以是个张量，也可以是一个数组或list。
tf.size(input,name=None) #返回一个张量，输入数据的元素数量。
tf.rank(input, name=None) #返回一个张量，输入数据的rank。
tf.reshape(input, shape, name=None) #将原有输入数据的shape按照指定形状进行变化，生成一个新的张量。
tf.expand_dims(input, dim, name=None) #插入维度1进行一个tensor中。
tf.squeeze(input,dim,name=None) #将dim指定的维度去掉（dim所指定的维度必须为1，否则出错）。
4、数据操作
tf.slice(input,begin,size,name=None) #将输入数据input进行切片操作，begin与size可以为list类型。
tf.split(value,num_or_size_splits, axis=0, num=None, name="split") #沿着某一维度将tensor分离为num_or_size_splits。
tf.concat(concat_dim, values, name='concat') #沿着某一维度连接tensor
tf.stack(input, axis=0) #将两个N维张量列表沿着axis轴组合成一个N+1维的张量
tf.unstack(value, num=None, axis=0, name="unstack") #将输入value按照指定的列或行进行拆分，并输出含有num个元素的列表（list）axis=0表示按行拆分，axis=1表示按列拆分。
tf.gather(params,indices,validate_indices=None,name=None) #合并索引indices所指定params中的切片
tf.one_hot(indices,depth,on_value=None,off_value=None,axis=None,dtype=None,name=None) #生成符合onehot编码的张量。
tf.count_nonzero(input_tensor,axis=None,keep_dims=False,dtype=dtype.int64,name=None,reduction_indices=None) #统计非0个数
5、算术运算函数
tf.assign(x,y,name=None) #令x=y
tf.add(x,y,name=None) #求和
tf.subtract(x,y,name=None) #减法
tf.multiply(x,y,name=None) #乘法
tf.divide(x,y,name=None) #除法
tf.mod(x,y,name=None) #取模
tf.abs(x,name=None) #求绝对值
tf.negative(x, name=None) #取负
tf.sign(x, name=None) #根据x的符号，返回0或1
tf.square(x,name=None) #平方
tf.round(x, name=None) #舍入最接近的整数。
tf.sqrt(x,name=None) #开根号
tf.pow(x,y,name=None) #幂次方计算
tf.exp(x,name=None) #计算e的次方
tf.log(x,name=None) #计算log，一输入计算e的ln，两输入以第二输入为底。
tf.maximum(x,y,name=None) #返回最大值
tf.minimum(x,y,name=None) #返回最小值
tf.cos(x,name=None) #三角函数sin,tan,atan
tf.cond(pred,true_fn=None,false_fn=None,strict=False,name=None,fn1=None,fn2=None) #满足条件执行fn1，否则执行fn2
6、矩阵相关运算
tf.diag(diagonal,name=None) #返回一个给定对角值的对角tensor。
tf.diag_part(input,name=None) #功能与上面相反
tf.trace(x,name=None) #求一个二维tensor足迹，即对角值diagonal之和
tf.transpose(a,perm=None,name='transpose') #让输入a按照参数perm指定的维度顺序进行转置操作。
tf.reverse(tensor,dims,name=None) #沿着指定的维度对输入进行反转。其中，dims为列表，元素含义为指向输入shape的索引。
tf.matmul(a,b,transpose_a=False,transpose_b=False) #矩阵相乘
tf.matrix_determinant(input,name=None) #返回方阵的行列式
tf.matrix_inverse(input,adjoint=None,name=None) #求方阵的逆矩阵
tf.cholesky(input,name=None) #对输入方阵cholesky分解，即把一个对称正定的矩阵表示成一个下三角矩阵L和其转置的乘积的分级
tf.matrix_solve(matrix,rhs,adjoint=None,name=None) #求解矩阵方程，返回矩阵变量。
7、序列比较与索引提取
tf.argmin(input,axis,name=None) #返回input最小值的索引index
tf.argmax(input,axis,name=None) #返回input最大值的索引index。
tf.setdiffld(x,y,name=None) #返回x,y中不同值的索引
tf.where(condition,x=None,y=None,name=None) #根据指定条件，返回对应的值或坐标。
tf.unique(x,name=None) #返回一个元组tuple(y,idx)。其中，y为x列表的唯一化数据列表，idx为x数据对应y元素的index。
tf.invert_permutation(x,name=None) #沿着input的第一维进行随机重新排列。
