tensorflow里面常用的函数

tf.argmax（）

argmax（）里面的第二个参数等于0，是表示已纵向的方式比较不同数组相应值中最大的，并返回其中最大的元素所在的索引

      [1,2,3]----------->0

      [2,3,4]----------->1

      [3,4,5]----------->2

      [4,5,6]----------->3

结果[3,3,3 ]

argmax（）里面的第二个参数等于1，是表示已横向的方式比较不同数值中最大的，并返回其中最大的元素所在的索引

      [1,2,3]----------->2

      [2,3,4]----------->2

      [3,4,5]----------->2

      [4,5,6]----------->2

结果[2,2,2,2]