![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch基础
文章平均质量分 77
日晨难再
未入行菜鸟
展开
-
Pytorch基础:Tensor的连续性
在Pytorch中,一个连续的张量指的是张量中各数据元素在底层的存储顺序与其在张量中的位置一致。这意味着每一个元素的地址可以通过下面的线性映射公式来确定:原创 2024-07-30 15:28:45 · 967 阅读 · 6 评论 -
Pytorch基础:Tensor的view方法(非连续张量也可以使用view)
在Pytorch中,view是Tensor的一个重要方法,用于返回一个改变了形状,但数据和数据的顺序与原来一致的新张量,但是新张量需要与原张量共用底层存储(称为视图)。原创 2024-07-27 18:33:19 · 1130 阅读 · 18 评论 -
Pytorch基础:Tensor的squeeze和unsqueeze方法
在Pytorch中,squeeze和unsqueeze是Tensor的一个重要方法,同时它们也是torch模块中的一个函数,它们的语法如下所示。原创 2024-07-26 21:32:43 · 667 阅读 · 4 评论 -
Pytorch:显卡驱动版本、Pytorch版本的关系
显卡驱动程序一般是显卡自带的,可以在终端中使用nvidia-smi命令查看,如下图所示。原创 2024-07-18 01:29:34 · 1207 阅读 · 14 评论 -
Pytorch基础:torch.cuda.set_device函数
torch.cuda.set_device函数用于设置当前使用的cuda设备,在当拥有多个可用的GPU且能被pytorch识别的cuda设备情况下(环境变量CUDA_VISIBLE_DEVICES可以影响GPU设备到cuda设备的映射)。原创 2024-05-13 17:06:23 · 1711 阅读 · 31 评论 -
Pytorch基础:环境变量CUDA_VISIBLE_DEVICES
CUDA_VISIBLE_DEVICES这个环境变量可以影响CUDA能识别到的GPU,并影响它映射到的cuda设备编号。原创 2024-05-13 00:00:15 · 1156 阅读 · 11 评论 -
Pytorch基础:torch.load_state_dict()方法在加载时不会检查类型
笔者在使用torch.nn.module的load_state_dict中出现了一个问题,一个被注册的张量在加载后居然没有变化,一开始以为是加载出现了问题,但发现其他参数加载成功,思索后发现是注册的张量的类型是整型而checkpoint中保存为浮点数类型,恰好注册时的默认值给的是0,而checkpoint中的浮点数又在0到1之间,因此出现了这个令人困惑的bug。原创 2024-04-27 20:58:20 · 1140 阅读 · 16 评论 -
Pytorch基础:张量相关的乘法(torch.mul、torch.mv、torch.matmul、torch.mm、torch.bmm、torch.dot、torch.tensordot、*、@)
Pytorch中含有很多种张量乘法,本文旨在帮助理解它们的不同。下面将分小节进行详细阐述,包括torch.mul、torch.matmul、torch.mm、torch.bmm、torch.dot、torch.tensordot、*、@。原创 2024-04-16 00:30:39 · 1349 阅读 · 24 评论 -
Pytorch基础:标量(零阶张量)
在PyTorch中,标量(Scalar)是一种只包含单个数值的零阶张量,它是只有一个元素的张量(注意:只有一个元素的张量并不一定是张量)。标量在深度学习中非常常见,通常用来表示损失函数的值、模型参数的值或其他单一数值。原创 2024-04-14 21:29:30 · 464 阅读 · 13 评论 -
Pytorch基础:Tensor的flatten方法
在Pytorch中,flatten是Tensor的一个重要方法,同时它也是一个torch模块中的一个函数,它们的语法如下所示。原创 2024-03-08 20:42:45 · 1561 阅读 · 17 评论 -
Pytorch基础:Tensor的permute方法
在Pytorch中,permute是Tensor的一个重要方法,同时它也是一个torch模块中的一个函数,它们的语法如下所示。官方的解释是:返回原始张量输入的视图,并对其维度进行转置。原创 2023-10-06 17:20:30 · 2305 阅读 · 10 评论 -
Pytorch基础:Tensor的transpose方法
在Pytorch中,transpose是Tensor类的一个重要方法,同时它也是torch模块中的一个函数,它们的语法如下所示。原创 2023-10-02 14:32:27 · 3188 阅读 · 1 评论 -
Pytorch基础:Tensor的reshape方法
在Pytorch中,reshape是Tensor的一个重要方法,它与Numpy中的reshape类似,用于返回一个改变了形状但数据和数据顺序和原来一致的新Tensor对象。注意:此时返回的新tensor中的数据对象并不一定是新的,这取决于应用此方法的Tensor是否是连续的。原创 2023-10-01 14:04:32 · 7787 阅读 · 11 评论