tensorflow 之 tf.argmax()解析

快来打我* 2021-11-23 09:38 554阅读 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 条评论，554人围观）

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

相关阅读

相关 TensorFlow 基础概念解析

• TensorFlow模块与架构介绍以下资料来源于极客时间学习资料 **TensorFlow 模块与 APIs** **![1604514-20190724...

太过爱你忘了你带给我的痛/ 2024年04月20日 09:13/ 0 赞/ 86 阅读

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

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

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

相关【tensorflow】关于tensorflow的losses.contrastive_loss解析

先说下目的吧，看图像检索相关的损失函数时，这个损失函数应该算是绕不开的前人工作。所以这里针对看到的一些文章，也做做自己的相关总结。主要是关于源码的学习啦。文章参考链接：

我就是我/ 2023年01月19日 07:06/ 0 赞/ 20 阅读

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

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

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

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

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

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

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

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

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

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

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

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

相关 SpringMvc之DispatchServlet解析

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

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

相关 tensorflow 之 tf.argmax()解析

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

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

相关 jsoup之页面解析

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

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