python topk()函数求最大和最小值实例


Posted in Python onApril 02, 2020

函数介绍

a.topk()求a中的最大值或最小值,返回两个值,一个是a中的值(最大或最小),一个是这个值的索引。

代码示例

>>> import torch
>>> a=torch.randn((3,5))
>>> a
tensor([[-0.4790, -0.6308, 0.2370, 0.0380, -0.0579],
  [-0.6712, -3.5483, -0.2370, -0.8658, 0.4145],
  [-1.4126, -0.8786, -0.4216, -0.0878, -1.4015]])
>>> _,pre=a.topk(1,dim=1,largest=True)
>>> pre
tensor([[2],
  [4],
  [3]])
>>> _
tensor([[ 0.2370],
  [ 0.4145],
  [-0.0878]])
>>> _,pre=a.topk(1,dim=1,largest=False)
>>> pre
tensor([[1],
  [1],
  [0]])
>>> _
tensor([[-0.6308],
  [-3.5483],
  [-1.4126]])

dim=1,为按行求最大最小值,largest为Ture,求最大值,largest=False,求最小值。

补充知识:python topN 取 最大的N个数 或 最小的N个数

来看代码吧!

import numpy as np
a = np.array([1,4,3,5,2])
b = np.argsort(a)
print(b)

print结果[0 4 2 1 3]

说明a[0]最小,a[3]最大

a[0]<a[4]<a[2]<a[1]<a[3]

以上这篇python topk()函数求最大和最小值实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中使用copy模块实现列表(list)拷贝
Apr 14 Python
Python编程中运用闭包时所需要注意的一些地方
May 02 Python
Python极简代码实现杨辉三角示例代码
Nov 15 Python
python函数的5种参数详解
Feb 24 Python
Python运维之获取系统CPU信息的实现方法
Jun 11 Python
python实现简单的单变量线性回归方法
Nov 08 Python
python 协程 gevent原理与用法分析
Nov 22 Python
pycharm解决关闭flask后依旧可以访问服务的问题
Apr 03 Python
Windows 下更改 jupyterlab 默认启动位置的教程详解
May 18 Python
pycharm软件实现设置自动保存操作
Jun 08 Python
windows+vscode安装paddleOCR运行环境的步骤
Nov 11 Python
Pytorch - TORCH.NN.INIT 参数初始化的操作
Feb 27 Python
python使用梯度下降和牛顿法寻找Rosenbrock函数最小值实例
Apr 02 #Python
Python如何通过百度翻译API实现翻译功能
Apr 02 #Python
Python基于百度AI实现OCR文字识别
Apr 02 #Python
python 穷举指定长度的密码例子
Apr 02 #Python
python3安装OCR识别库tesserocr过程图解
Apr 02 #Python
python简单的三元一次方程求解实例
Apr 02 #Python
Python 线性回归分析以及评价指标详解
Apr 02 #Python
You might like
php 数据库字段复用的基本原理与示例
2011/07/22 PHP
PHP对MongoDB[NoSQL]数据库的操作
2013/03/01 PHP
PHP+jQuery 注册模块的改进(三):更新到Smarty3.1
2014/10/14 PHP
js变量作用域及可访问性的探讨
2006/11/23 Javascript
表单填写时用回车代替TAB的实现方法
2007/10/09 Javascript
腾讯与新浪的通过IP地址获取当前地理位置(省份)的接口
2010/07/26 Javascript
JavaScript 语言基础知识点总结(思维导图)
2013/11/10 Javascript
JSON+HTML实现国家省市联动选择效果
2014/05/18 Javascript
AngularJS初始化过程分析(引导程序)
2014/12/06 Javascript
NodeJS使用jQuery选择器操作DOM
2015/02/13 NodeJs
多种jQuery绑定事件的实现方式
2016/06/13 Javascript
JS/jquery实现一个网页内同时调用多个倒计时的方法
2017/04/27 jQuery
AngularJS获取json数据的方法详解
2017/05/27 Javascript
ionic App问题总结系列之ionic点击系统返回键退出App
2017/08/19 Javascript
layer弹出层 iframe层去掉滚动条的实例代码
2018/08/17 Javascript
NodeJs之word文件生成与解析的实现代码
2019/04/01 NodeJs
Vue在chrome44偶现点击子元素事件无法冒泡的解决方法
2019/12/15 Javascript
vue-cli3项目配置eslint代码规范的完整步骤
2020/09/10 Javascript
python返回昨天日期的方法
2015/05/13 Python
Python 文件操作的详解及实例
2017/09/18 Python
python实现本地图片转存并重命名的示例代码
2018/10/27 Python
对python以16进制打印字节数组的方法详解
2019/01/24 Python
windows下python安装pip方法详解
2020/02/10 Python
css3圆角样式分享自定义按钮样式
2013/12/27 HTML / CSS
什么是Linux虚拟文件系统VFS
2012/01/31 面试题
心碎乌托邦的创业计划书范文
2013/12/26 职场文书
数控技术专科生自我评价
2014/01/08 职场文书
会计电算化个人求职信范文
2014/01/24 职场文书
二年级小学生评语
2014/04/21 职场文书
幼儿园家长安全责任书
2014/07/22 职场文书
党员教师四风自我剖析材料
2014/09/30 职场文书
小学校园广播稿集锦
2014/10/04 职场文书
皇城相府导游词
2015/02/06 职场文书
工程部主管岗位职责
2015/02/12 职场文书
Django项目如何获得SSL证书与配置HTTPS
2021/04/30 Python
python 如何执行控制台命令与操作剪切板
2021/05/20 Python