您好,欢迎来到刀刀网。
搜索
您的当前位置:首页PyTorch:Tensor的维度变换

PyTorch:Tensor的维度变换

来源:刀刀网

1.view  /  reshape

#Tensor的维度变换1:view和reshape,两者完全可以通用
a=torch.rand(4,1,28,28)#随机均匀初始化,四张28*28尺寸的灰度图片
print(a.shape)

print(a.view(4,28*28))#直接把位置信息和通道信息合并
print(a.view(4,28*28).shape)
print(a.view(4,28*28).dim())#2

print(a.view(4*28,28).shape)#把所有的照片的所有通道和所有行合并
print(a.view(4*28,28).dim())#2

print(a.view(4*1,28,28).shape)#把照片和通道合并
print(a.view(4*1,28,28).dim())#3

print(a.view(4,784).shape)#最后会恢复成[4,28,28,1],Logic Bug

2.unsqueeze : 增加一个维度

#Tensor的维度变换2:unsqueeze增加一个维度
b=torch.rand(4,1,28,28)
print("示例一:")
print(b.shape)#torch.Size([4, 1, 28, 28])

print("示例二:")#增加额外的维度,没有增加数据,增加一个组,一个组里面有四张图片
print(b.unsqueeze(0).shape)#torch.Size([1, 4, 1, 28, 28])

print("示例三:")
print(b.unsqueeze(-1).shape)#torch.Size([4, 1, 28, 28, 1])

print("示例四:")
print(b.unsqueeze(4).shape)#torch.Size([

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- gamedaodao.com 版权所有 湘ICP备2022005869号-6

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务