tensorflow函数——tf.argmax(x, axis)

tf.argmax(x, axis)
这个函数是衍生于numpy库中的numpy.argmax()。tf.argmax(x, axis) 函数返回的是张量x中元素最大值对应的index号(而不是元素最大值)

  • 参数x: 是一个函数的输入,是一个张量。例如a=[[1,2,3],[4,5,6],[7,8,9]] 这样的3X3的矩阵
  • 参数axis: 英文为“坐标轴”意思,它有2个值,当axis=0,表示要从每一列中寻找最大元素值对应的index号;当axis=1,表示从每一行中寻找最大元素值对应的index号

在python中可以试着验证一下:

import tensorflow as tf
a=[[1,0,3],[8,-3,6],[5,1,7]]
with tf.Session() as sess:
     print(sess.run(tf.argmax(a,1)))#按照行去查找每一行中最大值对应的index
     print(sess.run(tf.argmax(a,0)))#按照列去查找每一列中最大值对应的index

运行结果如下:
tensorflow函数——tf.argmax(x, axis)_第1张图片

你可能感兴趣的:(TensorFlow学习之路)