torch中一些代码小记

懿朔

于 2021-07-11 20:41:07 发布

阅读量232

点赞数

CC 4.0 BY-SA版权

分类专栏：代码相关问题

8 篇文章

订阅专栏

本文详细介绍了PyTorch中重塑张量的方法，包括reshape()和view()的区别。reshape()可以处理非连续性的tensor，而view()则适用于连续性tensor。此外，还讲解了unsqueeze用于增加维度和squeeze用于去除单个维度的操作，帮助理解张量形状变换的基本用法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

- torch中的维度改变 reshape() / view()

a=torch.ones([24,8,32])
b=a.reshape(-1,32)  
c=a.view(-1,32)
print(b.shape) #torch.Size([192, 32])
print(c.shape) #torch.Size([192, 32])

reshape() 和 view() 的区别
两者都是用来重塑tensor的shape的。view只适合对满足连续性条件（contiguous）的 tensor进行操作，而reshape同时还可以对不满足连续性条件的tensor进行操作，具有更好的鲁棒性。view能干的reshape都能干，如果view不能干就可以用reshape来处理。别看目录挺多，
reference：https://blog.youkuaiyun.com/Flag_ing/article/details/109129752

- torch中的维度扩展 unsqueeze

a=torch.ones([192,32])
a=a.unsqueeze(1)
print(a.shape)  
# torch.Size([192, 1, 32])

- 与 unsqueeze 对应的降维 squeeze

a=torch.ones([192,32])
a=a.unsqueeze(1)
print(a.shape)
# torch.Size([192, 1, 32]) 
a=a.squeeze(1)
print(a.shape)
# torch.Size([192, 32])