Pytorch
文章平均质量分 89
本栏主要是涉及用到pytorch的相关内容。
上海大学 吴昊
计算机专业博士研究生,华为云计算专家,CSDN博客专家,CSDN人工智能领域优质创作者,现主要研究Python数据分析、机器学习、NLP相关领域内容。
展开
-
Anaconda安装Pytorch(win系统)
前面有一篇博客专门讲了安装CPU版本的Pytorch,因为当时没有GPU,现在有了3090,专门记录一下安装GPU版的过程。原创 2023-03-06 22:19:13 · 693 阅读 · 0 评论 -
PyTorch的torch.cat、squeeze()、unsqueeze()和size()函数
目录 一、sequeeze()函数 二、unsequeeze()函数 三、size()函数 四、torch.cat函数 在Pytorch做深度学习过程中,CNN的卷积和池化过程中会用到torch.cat、squeeze()、unsqueeze()和size()函数,下面分别做讲解: 一、sequeeze()函数 x.squeeze(dim) 用途:进行维度压缩,去掉tensor中维数为1的维度 参数设置:如果设置dim=a,就是去掉指定维度中维数为1的 示例: import ..原创 2022-04-13 10:16:16 · 2509 阅读 · 0 评论