Tensorflow中 tf.argmax()方法的用法

tf.argmax()方法用于计算张量tensor沿着某一维度的最大值的索引。

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

第一个参数 input： 输入的需要处理的张量 tensor，必须是以下数据类型之一float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, uint16, complex128, half, uint32, uint64
第二个参数 axis： 指定的维（或轴），取值为0，代表延列取最大值索引，取值为1，表示延行取最大值索引，如果不指定，则默认为按照列计算，取值为2,3,4...是，则为多维函数的相应维度，以此类推
第三个参数 name： 操作的名称;
第四个参数 dimension：已弃用;
第五个参数 output_type：返回值的数据类型，是tf.int32, tf.int64之一。默认为