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去除列表中重复元素的方法
Mar 20 Python
Python列表list解析操作示例【整数操作、字符操作、矩阵操作】
Jul 25 Python
Python反射用法实例简析
Dec 22 Python
python print出共轭复数的方法详解
Jun 25 Python
python+selenium 鼠标事件操作方法
Aug 24 Python
关于Python3 类方法、静态方法新解
Aug 30 Python
python实现二分类的卡方分箱示例
Nov 22 Python
Python集成开发工具Pycharm的安装和使用详解
Mar 18 Python
python+django+selenium搭建简易自动化测试
Aug 19 Python
Python安装第三方库攻略(pip和Anaconda)
Oct 15 Python
Python类class参数self原理解析
Nov 19 Python
python SOCKET编程基础入门
Feb 27 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漏洞全解(详细介绍)
2012/11/13 PHP
PHP实现针对日期,月数,天数,周数,小时,分,秒等的加减运算示例【基于strtotime】
2017/04/19 PHP
js prototype截取字符串函数
2010/04/01 Javascript
dreamweaver 安装Jquery智能提示
2011/04/02 Javascript
js对列表中第一个值处理与jsp页面对列表中第一个值处理的区别详解
2013/11/05 Javascript
使用CSS样式position:fixed水平滚动的方法
2014/02/19 Javascript
jQuery学习笔记之总体架构
2014/06/03 Javascript
Bootstrap整体框架之CSS12栅格系统
2016/12/15 Javascript
Bootstrap的popover(弹出框)在append后弹不出(失效)
2017/02/27 Javascript
使用JS代码实现俄罗斯方块游戏
2018/08/03 Javascript
对layui中的onevent 和event的使用详解
2019/09/06 Javascript
html2canvas属性和使用方法以及如何使用html2canvas将HTML内容写入Canvas生成图片
2020/01/12 Javascript
Vue开发环境跨域访问问题
2020/01/22 Javascript
ES6 Symbol在对象中的作用实例分析
2020/06/06 Javascript
基于Python的身份证号码自动生成程序
2014/08/15 Python
实例说明Python中比较运算符的使用
2015/05/13 Python
利用Python进行异常值分析实例代码
2017/12/07 Python
使用python对文件中的数值进行累加的实例
2018/11/28 Python
Win10下Python3.7.3安装教程图解
2019/07/08 Python
python批量检查两个对应的txt文件的行数是否一致的实例代码
2020/10/31 Python
python日志通过不同的等级打印不同的颜色(示例代码)
2021/01/13 Python
5个你不知道的HTML5的接口介绍
2013/08/07 HTML / CSS
html5教程画矩形代码分享
2013/12/04 HTML / CSS
台湾森森购物网:U-mall
2017/10/16 全球购物
Notino芬兰:购买香水和化妆品
2019/04/15 全球购物
费用会计岗位职责
2014/01/01 职场文书
志愿者爱心公益活动策划方案
2014/09/15 职场文书
出纳试用期自我鉴定范文
2014/09/16 职场文书
指导教师推荐意见
2015/06/05 职场文书
诚信高考倡议书
2019/06/24 职场文书
日本读研:怎样写好一篇日本研究计划书?
2019/07/15 职场文书
Html5通过数据流方式播放视频的实现
2021/04/27 HTML / CSS
go设置多个GOPATH的方式
2021/05/05 Golang
浅谈Java实现分布式事务的三种方案
2021/06/11 Java/Android
微信小程序scroll-view不能左右滑动问题的解决方法
2021/07/09 Javascript
利用Python读取微信朋友圈的多种方法总结
2021/08/23 Python