平常电灯胆

文章
2
资源
0
加入时间
3年1月13天

tf.argmax()的详细用法

tf.argmax(data, axis=None)用tensorflow 做 mnist分类时,用到这个接口,于是就研究了下这个接口的用法:如果是一维数组呢?data = tf.constant([1,2,3])with tf.Session() as sess:print(sess.run(tf.argmax(data, 0)))#轴默认为0print(sess.run(tf.argmax(data)))22这个很好理解,因为data是一维数组,axis只能为0(如果是1就会报