【Tensorflow】tf.argmax函数

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

此函数是对矩阵按行或列计算最大值

 

参数

• input：输入Tensor
• axis：0表示按列，1表示按行
• name：名称
• dimension：和axis功能一样，默认axis取值优先。新加的字段

返回：Tensor  一般是行或列的最大值下标向量

 

例：