[python] tensorflow中的argmax()函数argmax()函数
2023-04-18 12:40:09 时间
首先 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]最大)
相关文章
- 【技术种草】cdn+轻量服务器+hugo=让博客“云原生”一下
- CLB运维&运营最佳实践 ---访问日志大洞察
- vnc方式登陆服务器
- 轻松学排序算法:眼睛直观感受几种常用排序算法
- 十二个经典的大数据项目
- 为什么使用 CDN 内容分发网络?
- 大数据——大数据默认端口号列表
- Weld 1.1.5.Final,JSR-299 的框架
- JavaFX 2012:彻底开源
- 提升as3程序性能的十大要点
- 通过凸面几何学进行独立于边际的在线多类学习
- 利用行动影响的规律性和部分已知的模型进行离线强化学习
- ModelLight:基于模型的交通信号控制的元强化学习
- 浅谈Visual Source Safe项目分支
- 基于先验知识的递归卡尔曼滤波的代理人联合状态和输入估计
- 结合网络结构和非线性恢复来提高声誉评估的性能
- 最佳实践丨云开发CloudBase多环境管理实践
- TimeVAE:用于生成多变量时间序列的变异自动编码器
- 具有线性阈值激活的神经网络:结构和算法
- 内网渗透之横向移动 -- 从域外向域内进行密码喷洒攻击