tensorflow基础（2）-tf.argmax和axis

tf.argmax()

这个函数的主要功能是返回最大值所在的坐标。主要用在分类的时候，如果只是简单的输出，只是对于每一类可能性的预测的输出，但是我要要的输出必须是确定的哪一类，所以需要确定里面的最大的值（也就是说最可能是哪一类）。

tf.argmax(input, axis=None, name=None, dimension=None, output_type=tf.int64)

input：一个张量(tensor),它的类型必须是一下的形式：

`float32`, `float64`, `int64`, `int32`, `uint8`, `uint16`, `int16`, `int8`, `complex64`, `complex128`, `qint8`, `quint8`, `qint32`, `half`.

axis：

参考博客是numpy模块之axis

博客讲解的非常的好，其实只要知道axis=0,1其实就够用了。axis=0代表列，axis代表行(我看了一下笔记，莫凡是这样说的，或许是为了好记)。哈哈，可能是错的。

例子

import tensorflow as tf

array = [[1,2,3],
        [4,5,6]]
with tf.Session() as sess:
    print(sess.run(tf.argmax(array,0)))
    print(sess.run(tf.argmax(array,1)))

输出

[1 1 1]
[2 2]