tf.arg_max和tf.argmax

参考官方文档感觉这两个函数作用差不多但是我习惯用tf.argmax

format:argmax(input, axis=None, name=None, dimension=None)

Args:input: A `Tensor`. Must be one of the following types: `float32`, `float64`, `int64`, `int32`, `uint8`, `uint16`, `int16`, `int8`, `complex64`, `complex128`, `qint8`, `quint8`, `qint32`, `half`.
      axis: A `Tensor`. Must be one of the following types: `int32`, `int64`.(注意这里是整形就行了)
        int32, 0 <= axis < rank(input).  Describes which axis
        of the input Tensor to reduce across. For vectors, use axis = 0.
      name: A name for the operation (optional).

栗子

import tensorflow as tf  
x=tf.constant([[1.,2.,6],[6.,2.,6]])  

xShape=tf.shape(x)
z1=tf.arg_max(x,1)#沿axis=0操作


with tf.Session() as sess:
    xShapeValue,d1=sess.run([xShape,z1])
    print('shape= %s'%(xShapeValue))
    print(d1)

结果


一般用来求一个矩阵中,每行最大值的index,一般用在测试的时候。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值