tf.get_collection("")#从集合中取全部变量,并生成一个列表
tf.add_n([])#列表中对应元素相加
tf.cast(x , dtype)#将x转为dtype类型
tf.argmax(x , axis)#(x是一个列表)返回最大值所在的索引号,如:tf.argmax([1 , 0 , 0] , 1),返回0;第二个参数axis(1)表示仅在第一个维度选取最大值(拿到每行的最大值)
os.path.join("home" , "name")#使用前要import os模块,它把字符串按路径形式拼接,本例返回"home/name"
'./model/mnist_model-1001'.split('/')[-1].split('-')[-1]#按指定拆分符对字符串拆分,返回拆分后的列表。本例中:先按'/'拆分,提取到最后一个元素'mnist_model-1001';再按'-'拆分,提取到最后一个元素'1001'
with tf.Graph().as_default() as g:#表示在其中定义的节点,在计算图G中
TensorFlow函数
最新推荐文章于 2024-05-15 05:06:21 发布