当前位置: 首页 > news >正文

pytorch学习过程中一些基础语法

1、tensor.view()函数,通俗理解就是reshape,#参数这里的-1需要注意,可以根据原张量size自行计算

data1=torch.randn((4,2))
data2=data1.view(2,4)
data3=data2.view(-1,8)

2、tensor.max()函数,在分类问题中,通常需要使用max()函数对softmax函数的输出值进行操作,求出预测值索引,然后与标签进行比对,计算准确率。
output = torch.max(input, dim)
输入:
input是一个tensor
dim是max函数索引的维度0/1,0是每列的最大值,1是每行的最大值
输出:
函数会返回两个tensor,第一个tensor是每行/列的最大值;第二个tensor是每行/列最大值的索引。
3、tensor equal 张量比较
equal:比较两个张量是否相等–相等返回:True; 否则返回:False
eq:比较两个张量tensor中,每一个对应位置上元素是否相等–对应位置相等,就返回一个True;否则返回一个False.
eq_:比较两个张量tensor中,每一个对应位置上元素是否相等–对应位置相等,就返回一个1;否则返回一个0. 并替换原张量

x = torch.tensor([1, 3, 5])
y = torch.tensor([1, 3, 7])
print(x.equal(y))
print(x.eq(y))
print(x.eq_(y))
print(x)
#输出
#False
#tensor([ True,  True, False])
#tensor([1, 1, 0])
#tensor([1, 1, 0])

4、tensor.squeeze()和tensor.unsqueeze()函数
tensor.unsqueeze(z),就是指在原有维度的x前面再加一个维度。就是扩充(维度增加,升维)
tensor.squeeze(z),就是指在维度为x的位置前减少一个维度。就是压缩(维度减少,降维)
只有某个维度是1,这个维度才能被压缩

import torch
data=torch.randn((3,4))
print(data.size())
data_up=data.unsqueeze(0)
print(data_up.size())
data_down=data_up.squeeze(0)
print(data_down.size())
#输出
#torch.Size([3, 4])
#torch.Size([1, 3, 4])
#torch.Size([3, 4])
http://www.lryc.cn/news/149438.html

相关文章:

  • 判断聚类 n_clusters
  • 基于深度学习的网络异常检测方法研究
  • SSM 基于注解的整合实现
  • 工具类APP如何解决黏性差、停留短、打开率低等痛点?
  • 使用Java MVC开发高效、可扩展的Web应用
  • wandb安装方法及本地部署教程
  • stable diffusion实践操作-提示词插件安装与使用
  • 【SpringBoot】详细介绍SpringBoot中的bean
  • 【Nuxt实战】在Nuxt3项目中如何按需引入Element-plus
  • 专业制造一体化ERP系统,专注于制造工厂生产管理信息化,可定制-亿发
  • Linux工具
  • Java项目-苍穹外卖-Day07-redis缓存应用-SpringCache/购物车功能
  • 零知识证明(zk-SNARK)(一)
  • linux中打印数据的行缓冲模式
  • 香橙派OrangePi zero H2+ 驱动移远4G/5G模块
  • 自动驾驶——【规划】记忆泊车特殊学习路径拟合
  • 【跟小嘉学 Rust 编程】十六、无畏并发(Fearless Concurrency)
  • Android 进阶——图形显示系统之VSync和 Choreographer的创建详解(一)
  • SQL Server开启变更数据捕获(CDC)
  • 八、性能测试
  • 景芯SoC 芯片全流程培训
  • 目标检测后的图像上绘制边界框和标签
  • Leetcode: 1. 两数之和 【题解超详细】
  • PHP 通过 Redis 解决并发请求的操作问题
  • 浅谈信息论和信息编码
  • 【测试】笔试02
  • 公司内部网段多管控乱,该如何规范跨网文件传输交换?
  • Ceph入门到精通-OSD waring 设置建议
  • 软件测试工程师如何快速理解业务?
  • 【教程】部署apprtc服务中安装google-cloud-cli组件的问题及解决