pytorch 常用函数 max ,eq说明


Posted in Python onJune 28, 2020

max找出tensor 的行或者列最大的值:

找出每行的最大值:

import torch
outputs=torch.FloatTensor([[1],[2],[3]])
print(torch.max(outputs.data,1))

输出:

(tensor([ 1., 2., 3.]), tensor([ 0, 0, 0]))

找出每列的最大值:

import torch
outputs=torch.FloatTensor([[1],[2],[3]])
print(torch.max(outputs.data,0))

输出结果:

(tensor([ 3.]), tensor([ 2]))

Tensor比较eq相等:

import torch

outputs=torch.FloatTensor([[1],[2],[3]])
targets=torch.FloatTensor([[0],[2],[3]])
print(targets.eq(outputs.data))

输出结果:

tensor([[ 0],
[ 1],
[ 1]], dtype=torch.uint8)

使用sum() 统计相等的个数:

import torch

outputs=torch.FloatTensor([[1],[2],[3]])
targets=torch.FloatTensor([[0],[2],[3]])
print(targets.eq(outputs.data).cpu().sum())

输出结果:

tensor(2)

补充知识:PyTorch - torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le

flyfish

torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le

以上全是简写

参数是input, other, out=None

逐元素比较input和other

返回是torch.BoolTensor

pytorch 常用函数 max ,eq说明

import torch

a=torch.tensor([[1, 2], [3, 4]])
b=torch.tensor([[1, 2], [4, 3]])

print(torch.eq(a,b))#equals
# tensor([[ True, True],
#     [False, False]])

print(torch.ne(a,b))#not equal to
# tensor([[False, False],
#     [ True, True]])

print(torch.gt(a,b))#greater than
# tensor([[False, False],
#     [False, True]])

print(torch.lt(a,b))#less than
# tensor([[False, False],
#     [ True, False]])

print(torch.ge(a,b))#greater than or equal to
# tensor([[ True, True],
#     [False, True]])

print(torch.le(a,b))#less than or equal to
# tensor([[ True, True],
#     [ True, False]])

以上这篇pytorch 常用函数 max ,eq说明就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python获取mp3文件信息的方法
Jun 15 Python
Python获取CPU、内存使用率以及网络使用状态代码
Feb 08 Python
Python3利用Dlib实现摄像头实时人脸检测和平铺显示示例
Feb 21 Python
Python学习笔记之图片人脸检测识别实例教程
Mar 06 Python
python3使用matplotlib绘制散点图
Mar 19 Python
Python3自定义json逐层解析器代码
May 11 Python
python给视频添加背景音乐并改变音量的具体方法
Jul 19 Python
如何使用 Flask 做一个评论系统
Nov 27 Python
python 爬取小说并下载的示例
Dec 07 Python
python的scipy.stats模块中正态分布常用函数总结
Feb 19 Python
Python 中的Sympy详细使用
Aug 07 Python
ubuntu安装jupyter并设置远程访问的实现
Mar 31 Python
浅谈pytorch中torch.max和F.softmax函数的维度解释
Jun 28 #Python
Python turtle库的画笔控制说明
Jun 28 #Python
使用python修改文件并立即写回到原始位置操作(inplace读写)
Jun 28 #Python
python删除指定列或多列单个或多个内容实例
Jun 28 #Python
Python3实现建造者模式的示例代码
Jun 28 #Python
Python工程师必考的6个经典面试题
Jun 28 #Python
Python drop方法删除列之inplace参数实例
Jun 27 #Python
You might like
PHP+jQuery 注册模块的改进(一):验证码存入SESSION
2014/10/14 PHP
PHP实现图片自动清理的方法
2015/07/08 PHP
php文件系统处理方法小结
2016/05/23 PHP
PHP异步进程助手async-helper
2018/02/05 PHP
PHP使用ActiveMQ实现消息队列的方法详解
2019/05/31 PHP
PHP哈希表实现算法原理解析
2020/12/11 PHP
JQuery-tableDnD 拖拽的基本使用介绍
2013/07/04 Javascript
js父页面中使用子页面的方法
2016/01/09 Javascript
Jquery EasyUI实现treegrid上显示checkbox并取选定值的方法
2016/04/29 Javascript
vue从使用到源码实现教程详解
2016/09/19 Javascript
微信小程序 图片等比例缩放(图片自适应屏幕)
2016/11/16 Javascript
jQuery中on方法使用注意事项详解
2017/02/15 Javascript
JS库之Particles.js中文开发手册及参数详解
2017/09/13 Javascript
jquery 实现拖动文件上传加载进度条功能
2018/03/18 jQuery
Vue.js图片预览插件使用详解
2018/08/27 Javascript
ES6中字符串的使用方法扩展
2019/06/04 Javascript
如何基于JavaScript判断图片是否加载完成
2019/12/28 Javascript
[42:06]2019国际邀请赛全明星赛 8.23
2019/09/05 DOTA
Python中使用platform模块获取系统信息的用法教程
2016/07/08 Python
Python双精度浮点数运算并分行显示操作示例
2017/07/21 Python
django mysql数据库及图片上传接口详解
2019/07/18 Python
安装python依赖包psycopg2来调用postgresql的操作
2021/01/01 Python
无需JS和jQuery代码实现CSS3鼠标浮动放大图片
2016/11/21 HTML / CSS
澳大利亚领先的美容护肤品零售商之一:SkincareStore
2018/01/22 全球购物
德国内衣、泳装和睡衣网上商店:Bigsize Dessous
2018/07/09 全球购物
澳大利亚领先的亚麻品牌:Bed Threads
2019/12/16 全球购物
技术总监的工作职责
2013/11/13 职场文书
网上签名寄语活动留言
2014/01/18 职场文书
道路交通安全实施方案
2014/03/12 职场文书
农村面貌改造提升实施方案
2014/03/18 职场文书
农村党员对照检查材料
2014/09/24 职场文书
先进教师个人事迹材料
2014/12/15 职场文书
个人催款函范文
2015/06/23 职场文书
2016年八一建军节活动总结
2016/04/05 职场文书
关于python类SortedList详解
2021/09/04 Python
Win11显卡控制面板打开显卡设置方法
2022/04/20 数码科技