原型:
tf.argmax(array,axis)
用法: 针对传入函数的axis参数,去选取array中相对应轴元素值大的索引!
输入: 一个向量或者一个矩阵或者一个张量
返回: vector中的最大值的索引号,如果vector是一个向量,那就返回一个值,如果是一个矩阵,那就返回一个向量,这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。
原型:
tf.argmax(array,axis)
用法: 针对传入函数的axis参数,去选取array中相对应轴元素值大的索引!
输入: 一个向量或者一个矩阵或者一个张量
返回: vector中的最大值的索引号,如果vector是一个向量,那就返回一个值,如果是一个矩阵,那就返回一个向量,这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。