深度学习——pytorch基础入门

一、张量

在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()函数来拆分张量。

二、

相关推荐
TTGGGFF13 分钟前
机器视觉:智能车大赛视觉组技术文档——用 YOLO3 Nano 实现目标检测并部署到 OpenART
人工智能·目标检测·计算机视觉
CodeJourney.18 分钟前
Python开发可视化音乐播放器教程(附代码)
数据库·人工智能·python
强德亨上校20 分钟前
神经网络详解
人工智能·深度学习·神经网络
视***间35 分钟前
275TOPS算力边缘计算盒子的价值洞察与市场定位---视程空间
人工智能·边缘计算
AI模块工坊35 分钟前
AAAI 2025 | 即插即用,川大Mesorch刷新SOTA,用「介观」Transformer架构终结图像造假
人工智能·深度学习·计算机视觉·架构·transformer
周杰伦_Jay37 分钟前
【OpenManus深度解析】MetaGPT团队打造的开源AI智能体框架,打破Manus闭源壁垒。包括架构分层、关键技术特点等内容
人工智能·深度学习·opencv·架构·开源
文火冰糖的硅基工坊1 小时前
[嵌入式系统-146]:五次工业革命对应的机器人形态的演进、主要功能的演进以及操作系统的演进
前端·网络·人工智能·嵌入式硬件·机器人
猫头虎1 小时前
openAI发布的AI浏览器:什么是Atlas?(含 ChatGPT 浏览功能)macOS 离线下载安装Atlas完整教程
人工智能·macos·chatgpt·langchain·prompt·aigc·agi
老六哥_AI助理指南1 小时前
为什么AI会改变单片机的未来?
人工智能·单片机·嵌入式硬件
SEO_juper1 小时前
2026 AI可见性:构建未来-proof策略的顶级工具
人工智能·搜索引擎·百度·工具·数字营销