pytorch 常用函数 max ,eq说明
作者:蓝鲸123 发布时间:2023-06-25 11:09:21
标签:pytorch,函数,max,eq
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
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]])
来源:https://blog.csdn.net/TH_NUM/article/details/80783037


猜你喜欢
- 前言 大家好,好男人就是我,我就是好男人,我就是-0nise。在各大漏洞举报平台,我们时常会
- 1.为conda配置清华源打开cmd输入以下命令:conda config --add channels https://mirrors.t
- 目录问题描述解决方案问题分析问题描述在 MySQL 中使用 delete 语句删除数据之后,监控视图中可用的磁盘空间没有增加,磁盘使用率没有
- 一个大于1的自然数,除了1和它本身外,不能被其他自然数(质数)整除(2, 3, 5, 7等),换句话说就是该数除了1和它本身以外不再有其他的
- 开启xp_cmdshell: exec sp_configure 'show advance
- 当数据库服务器变得十分繁忙导致性能下降时,你会怎么办?购买更多的硬件升级你的服务器,还是重新考虑数据库服务器设计使得数据库平台具备良好的可升
- //重新封装document对象 var Console={ Write:function(msg){alert(msg);} }; //P
- 引言:我们在页面上会看到,谁登录的就会显示谁的信息,那么这个页面上的变量信息是怎样实现的呢?这就是本文要讲述的内容——Django中的模板变
- 两种情况:1.带索引 2.不带索引前提介绍:方式:采用命令行的方式来模拟1.mysq由于默认是开启自动提交事务,所以首先得查看自己当前的数据
- 1、简单的按钮js事件 用于判断和显示提示 <script type="text/javascript&
- 1.1.1 摘要 Join是关系型数据库系统的重要操作之一,SQL Server中包含的常用Join:内联接、外联接和交叉联接等。如果我们想
- 用Python写个注册登录功能,供大家参考,具体内容如下本文是用Python写一个注册登录功能,难度不大,很适合练手主要就是用列表和字典,以
- 在实际应用中,我们经常需要使用定时器去触发一些事件。Python中通过线程实现定时器timer,其使用非常简单。看示例:import thr
- SQL Server具有强大的复制功能,除了将数据和数据库对象从一个数据库复制并准确分发的另一个数据库中,还要实行数据库之间的同步。SQL
- 从github上下载,链接为:https://github.com/jhao104/proxy_pool下载好之后解压文件,然后将文件夹目录
- 首先,为什么选择Tornado:1.高性能的网络库,这可以和gevent,twisted,libevent等做对。提供了异步io支持,超时事
- 测试的过程中经常需要截取屏幕,通常的做法是使用手机自带的截屏功能,然后将截屏文件复制出来,这种方法的优点是不需要连接数据线就可截屏,缺点则是
- 本文实例讲述了Python实现PS滤镜特效之扇形变换效果。分享给大家供大家参考,具体如下:这里用 Python 实现 PS 滤镜中的一种几何
- MySQL存储过程与存储函数的相关概念存储函数和存储过程的主要区别:存储函数一定会有返回值的存储过程不一定有返回值存储过程和函数能后将复杂的
- Css tranistions允许元素的属性在单位时间内发生平滑的过渡,在阅读完《CSS Transitions Module Level