
pytorch
pytorch安装以及相关操作指令、error解决等内容
X CODE
闻道有先后。
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
查看Pytorch版本
【代码】查看Pytorch版本。原创 2025-03-12 09:05:35 · 181 阅读 · 0 评论 -
CUDA版本查看
参考https://blog.51cto.com/u_16175435/12699176感谢大佬们的分享,本博客只用作学习笔记,侵删!原创 2025-03-12 09:01:49 · 294 阅读 · 0 评论 -
flatten()
python里的flatten(dim)表示,从第dim个维度开始展开,将后面的维度转化为一维.也就是说,只保留dim之前的维度,其他维度的数据全都挤在dim这一维。原创 2024-09-28 09:50:16 · 304 阅读 · 0 评论 -
torch.cat()
【代码】torch.cat()原创 2024-09-27 16:42:26 · 262 阅读 · 0 评论 -
expand()
将张量广播到新的形状。原创 2024-09-27 16:37:12 · 204 阅读 · 0 评论 -
nn.Parameter()
创建可训练的参数(权重和偏置),这些参数会在模型训练过程中自动更新。原创 2024-09-27 15:10:44 · 125 阅读 · 0 评论 -
torch.zeros()
【代码】torch.zeros()原创 2024-09-27 15:09:50 · 158 阅读 · 0 评论 -
.flatten()
从dim维度开始以后的维度都变成一维的。原创 2024-09-26 23:17:01 · 333 阅读 · 0 评论 -
nn.Conv2d()
【代码】nn.Conv2d()原创 2024-09-26 22:55:18 · 155 阅读 · 0 评论 -
【Pytorch】模型权重保存与上传
【代码】【Pytorch】模型权重保存与上传。原创 2024-08-29 22:12:23 · 472 阅读 · 0 评论 -
【Pytorch】torch.sort()
对输入数据排序,返回两个值,即排序后的数据values和其在原矩阵中的坐标indices。原创 2024-08-15 21:20:47 · 144 阅读 · 0 评论 -
【Pytorch】nn.Sequential()
允许通过按顺序堆叠多个层来创建神经网络模型。原创 2024-08-15 21:00:04 · 157 阅读 · 0 评论 -
【Pytorch】nn.Identity()
nn.Identity() 是 PyTorch 中的一个层(layer)。它实际上是一个恒等映射,不对输入进行任何变换或操作,只是简单地将输入返回作为输出。原创 2024-08-15 20:23:58 · 292 阅读 · 0 评论 -
【解决error】安装torch 1.1.0
【代码】【解决error】安装torch 1.1.0。原创 2024-08-04 22:01:08 · 439 阅读 · 0 评论 -
【Pytorch】torch.cat()函数
用于在指定的维度上拼接tensor(张量)。原创 2024-04-18 19:45:25 · 307 阅读 · 0 评论 -
【解决error】pytorch中使用torch.utils.data.TensorDataset时报错TypeError: ‘int‘ object is not callable
原代码:解决方法传入的参数必须是tensor类型的,所以把x_train与y_train转换为tensor类型在执行这个函数就不报错了,修改后的代码:参考博客https://www.cnblogs.com/wyhluckdog/p/11842285.html感谢大佬的分享,本博客只用作学习笔记,侵删!原创 2024-03-26 21:30:48 · 479 阅读 · 0 评论 -
【pytorch】安装合集
https://pytorch.org/get-started/previous-versions/原创 2024-03-26 11:00:34 · 303 阅读 · 0 评论