[python] tensorflow中的argmax()函数argmax()函数
2024-10-20 16:30:42
首先 import tensorflow as tf
tf.argmax(tenso,n)函数会返回tensor中参数指定的维度中的最大值的索引或者向量。当tensor为矩阵返回向量,tensor为向量返回索引号。其中n表示具体参数的维度。
以实际例子为说明:
import tensorflow as tf
import numpy as np
test = [[1, 6, 7], [2, 5, 8], [3, 4, 9]]
with tf.Session() as sess:
print(sess.run(tf.argmax(test, 0))) #输出:array([2, 0, 2]
print(sess.run(tf.argmax(test, 1))) #输出:array([2, 2, 2]
n=0代表列的比较,如下:
print(sess.run(tf.argmax(test, 0))) 指的是
test[0] = array([1, 6, 7])
test[1] = array([2, 5, 8])
test[2] = array([3, 4, 9])
#[2, 0, 2]=(第一列test[0][1]最大,第二列test[2][1]最大,第三列test[2][2]最大)
n=1代表行的比较,如下:
print(sess.run(tf.argmax(test, 1))) 指的是
test[0] = array([1, 6, 7]) #[2]=(第一行test[0][2]最大)
test[1] = array([2, 5, 8]) #[2]=(第二行test[1][2]最大)
test[2] = array([3, 4, 9]) #[2]=(第三行test[2][2]最大)
最新文章
- Redis主-从部署实践
- PHP实现发红包程序
- java.lang.reflect.Constructor
- 《JavaScript高级程序设计》chapter 1: javascript 简介
- iOS端给unity发送消息,实现两者交互。
- mysql概要(十三)备份和恢复
- SQL Server2008 程序设计 汇总 GROUP BY,WITH ROLLUP,WITH CUBE,GROUPING SETS(..)
- Java 并发——多线程基础
- 刷新的时候jquery获取checkbox是否为选中和设置选中
- 注册界面设计及实现之(三)SharedPerferences实现数据暂存
- qcow2、raw、vmdk等镜像格式
- c/c++常用的几个关键字总结
- 在CentOS 7.3 中安装 NVIDIA GT730 显卡驱动
- Spark-streaming 连接flume
- 【面试】我是如何在面试别人Spring事务时“套路”对方的
- 日期求星期(java)-蓝桥杯
- for循环比较
- mongodb shell 运行js脚本的四种方式
- Nginx+Tomcat+Https 服务器负载均衡配置
- 《Java程序设计》 第五周学习总结
热门文章
- Vu3.x如何给v-for循环出来的输入框绑定v-mode的值,以及实现父子组件传值、双向绑定
- ULID规范解读与实现原理
- python创建icon图标
- 分清国内版FireFox和国际版FireFox
- flutter 系列之:flutter 中的幽灵offstage
- Codeforces Round #812 (Div. 2) D. Tournament Countdown(交互题)
- android_studio 使用
- 「浙江理工大学ACM入队200题系列」问题 E: 零基础学C/C++78——求奇数的乘积
- Oracle 表空间常用操作
- 【JAVA】详解在JAVA中int与Integer的区别以及背后的原因。