Tensorflow模型操作常用函数总结

面将函数模型操作常用函数总结。如表所示:

表2 模型操作相关函数

函数

tf.train.Saver(var_list=None,reshape=False,

sharded=False,max_to_keep=5,

keep_checkpoint_every_n_hours=10000.0,

name=None,restore_sequentially=False,

saver_def=None,builder=None)

创建存储器Saver

tf.train.Saver.save(sess,save_path,global_step=None,

latest_filename=None,meta_graph_suffix=’meta’,

write_meta_graph=True)

保存

tf.train.Saver.restore(sess,save_path)

恢复

tf.train.Saver.last_checkpoints

列出最近未删除的checkpoint文件名

tf.train.Saver.set_last_checkpoints(last_checkpoints)

设置checkpoint文件名列表

tf.train.Saver.set_last_checkpoints_with_time(

last_checkpoints_with_time)

设置checkpoint文件名列表和时间戳

tf.ones_like (input)

生成和输出张量一样形状和类型的0。例如:

tensor=[[1,2,3],[4,5,6]]

tf.ones_like(tensor) ==>[[0 0 0][0 0 0]]

tf.zeros_like (input)

生成和输出张量一样形状和类型的1。例如:

tensor=[[1,2,3],[4,5,6]]

tf.zeros_like(tensor) ==>[[0 0 0][0 0 0]]

tf.fill(shape,value)

为指定形状填值。例如:

tf.fill([2,3],1)==>[[1 1 1][1 1 ]]

tf.constant(value,shape)

生成常量。例如:

tf.constant(1,[2,3])==>[[1 1 1][1 1 1]]

tf.random_normal(shape,mean=0.0,sddev=1.0,

dtype=tf.float32,seed=None,name=None)

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

tf.truncated_normal(shape,mean=0.0,sddev=1.0,

dtype=tf.float32,seed=None,name=None)

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

tf.random_uniform(shape,mean=0.0,sddev=1.0,

dtype=tf.float32,seed=None,name=None)

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

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.linspace(start=1.0,stop=5.0,

num=5,name=None)

[1,2,3,4,5]

tf.range(start,limit=None,delta=1,name=’range’)

在[start,limit]范围内以步进值delta产生等差数列。注意,不包括limit在内得。例如:

tf.range(start=1,limit=5,delta=1)

[1 2 3 4]

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值