首页 > 代码库 > TFboy养成记 tf.cast,tf.argmax,tf.reduce_sum

TFboy养成记 tf.cast,tf.argmax,tf.reduce_sum

referrence: 莫烦视频

先介绍几个函数

1.tf.cast()

技术分享

英文解释:

技术分享

也就是说cast的直译,类似于映射,映射到一个你制定的类型。

 

2.tf.argmax 原型:

技术分享

含义:返回最大值所在的坐标。(谁给翻译下最后一句???)

技术分享

 ps:谁给解释下axis最后一句话?

 技术分享

例子:

技术分享

技术分享

 

 3.tf.reduce_mean() 原型:

技术分享

含义:一句话来说就是对制定的reduction_index进行均值计算。

技术分享

注意,reduction_indices为0时,是算的不同的[]的同一个位置上的均值

为1是是算的[]内的均值。

 1 import numpy as np
 2 import tensorflow as tf
 3 tt = tf.reduce_mean([1,2,3,4])
 4 ttt = tf.reduce_mean([[1.0,2,3,4,5],[6,7,8,9,10],[1,1,1,1,1]],reduction_indices=[1])
 5 t = tf.reduce_mean([[1.0,2,3,4,5],[6,7,8,9,10],[1,1,1,1,1]],reduction_indices=[0])
 6 sess = tf.Session()
 7 sess.run(tf.initialize_all_variables())
 8 
 9 print (sess.run(t))
10 print (sess.run(tt))

 

 技术分享

 

TFboy养成记 tf.cast,tf.argmax,tf.reduce_sum