tensorflow 之 tf.argmax()解析

快来打我* 2021-11-23 09:38 537阅读 0赞

运行环境 : python 3.6.0

第三方库 : tensorflow 1.9.0

> 用 tensorflow 做 CNN\_TEXT 文本分类时，看到这个API，然后去官网查了一下，再看了一下别的资料，算是明白它的处理方式了。

> tf.argmax 可以认为就是 np.argmax 。应该说 tensorflow 使用 numpy 实现的这个 API 。简单的说 ， tf.argmax 就是返回最大的那个值所对应的下标 。

tf.argmax 有个很重要的参数 axis ， 像这样 tf.argmax(test, 0) 和 tf.argmax(test, 0) 有什么区别呢？

看示例 :

# -*- encoding: utf-8 -*-
    
    import numpy as np
    
    test = np.array([
        [1, 2, 3],
        [2, 3, 4],
        [5, 4, 3],
        [8, 7, 2]])
    
    print("axis=0", np.argmax(test, 0))
    print("axis=1", np.argmax(test, 1))
    
    # 输出
    """
    axis=0 [3 3 1]
    axis=1 [2 2 0 0]
    """

有明白这是啥意思没 ?

> 答案是这样的 :
> 
> 当 axis=0 的时候 ， 这时候是求出 每一列 最大值 所对应的 索引数字 ， 然后 将 这几个数字 输出一个 列表 并返回
> 
> 当 axis=1 的时候 ， 这时候是求出 每一行 最大值 所对应的 索引数字 ， 然后 将 这几个数字 输出一个 列表 并返回
> 
>

*  axis = 0 :

![20190726121648811.png][]

*  axis = 1 :

![20190726121759395.png][]

> 以上情况为数组长度一致的情况 ， 如果数组禅古不一致 ， axis 最大值 为 最小的数组长度 -1 ， 超过则报错。   
> 当不一致的时候 ， axis = 0 的比较也就变成了每个数组的和的比较 。

[20190726121648811.png]: /images/20211122/740f10571d6b4509b44754fa84cddf2b.png
[20190726121759395.png]: /images/20211122/610f42c065c148058089c7ac495c76be.png

发表评论取消回复

表情：

评论列表（有 0 条评论，537人围观）

还没有评论，来说两句吧...

相关阅读

相关图注意力网络(GAT) TensorFlow解析

论文图注意力网络来自 Graph Attention Networks，ICLR 2018. https://arxiv.org/abs/1710.10903 注意力

﹏ヽ暗。殇╰゛Y/ 2023年07月19日 05:53/ 0 赞/ 1 阅读

相关 Android之Json解析

JSON返回示例： \{ /\JSONObject\/ "resultcode": "200", "reason": "SUCCESSED!",

雨点打透心脏的1/2处/ 2022年09月23日 04:51/ 0 赞/ 248 阅读

相关 Tensorflow深度学习入门（11）——Tensorflow代码解析

本文转自[http://mp.weixin.qq.com/s/wC2EKp1IBg5ow4lShUf5tA][http_mp.weixin.qq.com_s_wC2EKp1IB

男娘i/ 2022年06月16日 07:53/ 0 赞/ 377 阅读

相关 python-tensorflow(1)C3D代码解析

标签（空格分隔）：文章复现 -------------------- C3D代码解析 1、main函数 def main(_): run_tr

川长思鸟来/ 2022年06月08日 02:41/ 0 赞/ 229 阅读

相关 tensorflow(6) mnist.train.next_batch()函数解析

tensorflow的feed\_dict原理，它需要的每次生成一个batch的数据。 1. Dataset类将数据处理部分写成一个类，init函数中定义了一些参数

客官°小女子只卖身不卖艺/ 2022年06月04日 07:47/ 0 赞/ 229 阅读

相关 face_recognation人脸识别tensorflow版本代码解析

/\ 版权声明：可以任意转载，转载时请标明文章原始出处和作者信息 .\/ \\\人脸识别程序共分为4个部分，大致结构如下： \\\\1.get\_my\_faces\_d

╰半橙微兮°/ 2022年05月17日 00:04/ 0 赞/ 245 阅读

相关 SpringMvc之DispatchServlet解析

一.介绍 DispatcherServlet（前端控制器）是整个MVC框架中最为核心的一块，它主要用来拦截符合要求的外部请求，并把请求分发到不同的控制器去处理，根据控制器

野性酷女/ 2022年04月10日 07:20/ 0 赞/ 266 阅读

相关 Java之String解析

Java之String解析 1. String类 2. 字符串常量池 2.1 工作原理 2.2 实现前提 3. 实例 1.

缺乏、安全感/ 2022年03月22日 05:48/ 0 赞/ 242 阅读

相关 tensorflow 之 tf.argmax()解析

运行环境 : python 3.6.0 第三方库 : tensorflow 1.9.0 > 用 tensorflow 做 CNN\_TEXT 文本分类时，看到这个A

快来打我*/ 2021年11月23日 09:38/ 0 赞/ 538 阅读

相关 jsoup之页面解析

通过java不用正则表达式，实现对页面标签的过滤或解析。我们平时上网用的浏览器，除了根据用户输入的url完成通信，还要解析执行javascript代码浏览器在发送http

女爷i/ 2021年09月28日 02:36/ 0 赞/ 429 阅读