python topk()函数求最大和最小值实例
更新时间：2020年04月02日 10:00:13 作者：wang xiang
这篇文章主要介绍了python topk()函数求最大和最小值实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
函数介绍
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()函数求最大和最小值实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
通过pandas的使用，我们经常要交互式地展示表格（dataframe）、分析表格。本文整理了8个常用的配置选项，使用可以提高很多效率，需要的可以参考一下2022-06-06
这篇文章主要介绍了Python内置函数及功能简介汇总,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-10-10
这篇文章主要介绍了python 如何将字典写为json文件的操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-09-09
这篇文章主要介绍了python flask之模板继承方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-03-03
本篇文章主要介绍了python3.5实现socket通讯示例(TCP），小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-02-02
这篇文章主要介绍了pygame学习笔记（1）：矩形、圆型画图实例,本文讲解了pygame窗口、窗口退出、pygame中的颜色、圆形、矩形及一个完整实例,需要的朋友可以参考下2015-04-04
今天小编就为大家分享一篇YUV转为jpg图像的实现，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
Python简单检测文本类型的2种方法【基于文件头及cchardet库】
这篇文章主要介绍了Python简单检测文本类型的方法,结合实例形式分析了基于基于文件头及cchardet库两种文本类型检测的方法,需要的朋友可以参考下2016-09-09
这篇文章主要介绍了Python数据结构之哈夫曼树定义与使用方法,结合具体实例形式分析了Python哈夫曼树的原理、定义及简单使用方法,需要的朋友可以参考下2018-04-04
这篇文章主要介绍了TensorFlow实现简单的CNN的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-07-07
最新评论