TF-tf.arg_max 介绍

Posted .每天进步一点点

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了TF-tf.arg_max 介绍相关的知识,希望对你有一定的参考价值。

定义为

    def arg_max(input, dimension, name=None)

作用是取行或者列的最大值的位置。

input:
类型为 float32, float64, int64, int32, uint8, uint16, int16, int8, complex64, complex128, qint8, quint8, qint32, half的tensor

 

dimension:

必须为int32, int64. int32,取值为0或1.

 

name:
名字


returns:
返回一个tensor

例如以下测试

import tensorflow as tf

list_a = [[1,2,3,4,5],
          [3,3,3,1,6],
          [5,1,2,1,1]]

sess = tf.InteractiveSession()

argmax0 = tf.arg_max(list_a, 0)
print("argmax 0={}".format(argmax0.eval()))
argmax1 = tf.arg_max(list_a, 1)
print("argmax 1={}".format(argmax1.eval()))

结果为

argmax 0=[2 1 0 0 1]
argmax 1=[4 4 0]

 

以上是关于TF-tf.arg_max 介绍的主要内容,如果未能解决你的问题,请参考以下文章

VS code自定义用户代码片段snippet

Jacoco和Tycho surefire的Eclipse RCP插件代码介绍

使用 Git 来管理 Xcode 中的代码片段

48个值得掌握的JavaScript代码片段(上)

从零开始配置vim(27)——代码片段

Huawei_Netconf_Ncclient