pytorch学习笔记——张量维度操作
2021/9/23 23:10:52
本文主要是介绍pytorch学习笔记——张量维度操作,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
一、torch.cat()
添加链接描述
按照指定维度拼接张量
c=torch.cat((A,B),0)#按维度0拼接 c=torch.cat((A,B),1)#按维度1拼接
二、torch.view()
添加链接描述
x = torch.randn(4, 4) print(x.size()) y = x.view(16) print(y.size()) z = x.view(-1, 8) # -1表示该维度取决于其它维度大小,即(4*4)/ 8 print(z.size()) m = x.view(2, 2, 4) # 也可以变为更多维度 print(m.size()) a = torch.Tensor([[1, 2, 3], [4, 5, 6]]) # 定义一个 2*3 的 Tensor a = a.view(-1) print(a)
输出结果为:
torch.Size([4, 4]) torch.Size([16]) torch.Size([2, 8]) torch.Size([2, 2, 4]) tensor([1., 2., 3., 4., 5., 6.])
是pytorch中改变张量shape的函数,相当于numpy中的resize()和tensorflow中的reshape()。
三、torch.resize_
添加链接描述
改变张量的尺寸
四、torch.reshape
添加链接描述
五、torch.unsqueeze
unsqueeze()用来添加维度
squeeze()用来删除维度为一的维度或者删除指定维度唯一的维度
这篇关于pytorch学习笔记——张量维度操作的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-07-06有没有什么开源的py项目可以对图像进行分类-icode9专业技术文章分享
- 2024-07-05feign默认connecttimeout和readtimeout是多少-icode9专业技术文章分享
- 2024-07-05idea控制台,日志太多,导致部分想看得日志被刷走 搜不到-icode9专业技术文章分享
- 2024-07-05The server selected protocol version Tls10 is not accepted by client preferences [TLs12]-icode9专业技术文章分享
- 2024-07-05怎么清理项目缓存-icode9专业技术文章分享
- 2024-07-04安装 Eyoucms详细图文教程-icode9专业技术文章分享
- 2024-07-04ueditor 复制文章时,图片的链接是一个下载图片地址,该如何处理?-icode9专业技术文章分享
- 2024-07-04怎样判断host有没有对wordpress有缓存呢-icode9专业技术文章分享
- 2024-07-04具有编译功能的系统make后,无法ssh连接-icode9专业技术文章分享
- 2024-07-04make后如何升级ssh-icode9专业技术文章分享