深度学习——pytorch基础入门

embedded/2025/1/15 15:38:42/

一、张量

        在PyTorch中,张量是PyTorch中最基本的数据结构。张量可以看作是一个多维数组,可以在GPU上加速运算。PyTorch的张量和Numpy的数组非常类似,但是与Numpy不同的是,PyTorch的张量可以自动地在GPU上进行加速计算。

        PyTorch中的张量是深度学习模型中最常用的数据结构,它可以存储模型的输入、输出以及模型的参数,并且可以通过自动求导来计算梯度,方便进行反向传播算法的实现。

1、张量的数据类型

  • torch.float32torch.float: 默认的浮点数类型,32位浮点数。
  • torch.float64torch.double: 64位浮点数。
  • torch.float16torch.half: 半精度浮点数。适用于GPU计算,可以节省内存和计算资源。
  • torch.int8: 8位整数。
  • torch.int16torch.short: 16位整数。
  • torch.int32torch.int: 32位整数。
  • torch.int64torch.long: 64位整数。
  • torch.bool: 布尔类型,只能是True或False。

2、张量的创建

在PyTorch中,可以使用多种方式来创建张量。下面是一些常用的方法:

2.1、使用torch.tensor()函数:

        可以使用torch.tensor()函数从Python列表、元组、NumPy数组等数据结构中创建一个新的张量。例如:

         张量的维度可以通过.shape查看,并可使用.size0方法计算张量的形状大小,使用.numel0方法计算张量中包含元素的数量,例如:

        还可以使用参数dtype来指定张量的数据类型,使用参数requires_grad来指定张量是否需要计算梯度(只有浮点数才能计算梯度)。例如:

2.2、使用torch.Tensor()函数:

        在PyToreh中也可使用loreh.Tensor()函数来生成张量,而且可以根据指定的形状生成张量。

2.3、使用numpy数组转化为张量

        将Numpy数组转化为PyTorch张量,可以使用torch.as_tensor()兩数和torehrom_numpy()函数,例如:

2.4、随机数生成张量

        在PyToreh中还可以通过相关随机数来生成张量,并且可以指定生成随机数的分布函数等。在生成随机数之前,可以使用torch.manual_seed()函数,指定生成随机数的种子,用于保证生成的随机数是可重复出现的。

1. torch.randn():从标准正态分布中生成随机数。

2. torch.rand():从0到1之间的均匀分布中生成随机数。

3. torch.randint():在给定的范围内生成随机整数。

4.torch.normal():生成服从正态分布的随机数


2.5、其他

torch.zeros():全0张量
torch.ones():全1张量
torch.eye():单位张量
torch.full(():填充张量
torch.empty():空张量

3、张量操作

3.1、改变张量大小

        改变张量的形状在深度学习的使用过程中经常会遇到,而且针对不同的情况对张量形状尺寸的改变有多种函数和方法可以使用,如tensor.reshape()方法可以设置张量的形状大小。

3.2、获取张量元素

        该操作类似于python中数组的访问和切片 

3.3、拼接和拆分张量

        在PyTorch中,可以使用torch.cat()或torch.stack()函数来拼接张量,使用torch.split()torch.chunk()函数来拆分张量。

二、


http://www.ppmy.cn/embedded/153767.html

相关文章

嵌入式系统中的 OpenCV 与 OpenGLES 协同应用

🎬 秋野酱:《个人主页》 🔥 个人专栏:《Java专栏》《Python专栏》 ⛺️心若有所向往,何惧道阻且长 文章目录 一、OpenCV 在嵌入式中的基石地位二、OpenGLES 为嵌入式图形渲染赋能三、二者协同的精妙之处四、面临的挑战与应对策略 在嵌入式开…

STM32内置Flash

一、原理 利用flash存储用户数据需要注意查看,用户数据是否会覆盖芯片运行程序。 IAP(在程序中编程)利用程序修改程序本身,和OTA是一个原理。IAP在程序中编程支持任意一种通信下载。 ICP(在电路中编程,通…

Redis的哨兵机制

Redis 的主从复制模式下,一旦主节点由于故障不能提供服务,需要人工进行主从切换,同时大量的客户端需要被通知切换到新的主节点上,对于上了⼀定规模的应用来说,这种方案是无法接受的,于是 Redis 从 2.8 开始…

【AscendC】tiling方案设计不当引起的一个时隐时现的bug

在设计tiling方案时,通常会考虑到非对齐的场景,对输入数据进行补全操作从而使得非对齐场景也能正确的完成计算。但在某些算子的实现过程中,沿用上述操作却会造成数据的错误计算,且这种错误出现与否取决于随机生成的测试数据质量。…

要将Git仓库的master分支回滚到上一版本

要将Git仓库的master分支回滚到上一版本,你可以按照以下步骤操作: ‌确定要回滚的提交‌: 使用git log命令查看提交历史记录,找到你想要回滚到的上一版本的commit ID。 git log该命令将显示所有的提交记录,包括每个提…

有收到腾讯委托律师事务所向AppStore投诉带有【水印相机】主标题名称App的开发者吗

近期,有多名开发者反馈,收到来自腾讯科技 (深圳) 有限公司委托北京的一家**诚律师事务所卞,写给AppStore的投诉邮件。 邮件内容主要说的是,腾讯注册了【水印相机】这四个字的商标,所以你们这些在AppStore上的app&…

AIGC时代 | 探索AI Agent的奥秘:四种设计模式引领未来智能趋势

文章目录 1. 反思案例: 2. 工具使用案例: 3. 规划案例: 4. 多Agent协作案例: 总结作者简介:延伸阅读内容简介:目录 导读:AI Agent是指能够在特定环境中自主执行任务的人工智能系统,不…

基于微信小程序的电影交流平台设计与实现(LW+源码+讲解)

专注于大学生项目实战开发,讲解,毕业答疑辅导,欢迎高校老师/同行前辈交流合作✌。 技术范围:SpringBoot、Vue、SSM、HLMT、小程序、Jsp、PHP、Nodejs、Python、爬虫、数据可视化、安卓app、大数据、物联网、机器学习等设计与开发。 主要内容:…