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之一。默认为