pytorch tensor张量维度转换

原创于 2021-02-12 20:19:52 发布 · 262 阅读

0 ·

CC 4.0 BY-SA版权

PyTorch 专栏收录该内容

4 篇文章

订阅专栏

本文详细介绍了深度学习中常用的张量维度转换函数，包括view、reshape、permute、squeeze/unsqueeze、expand、narrow、resize_、repeat、unfold、cat和stack等，帮助理解它们的使用方法及应用场景，是深度学习初学者和实践者的必备参考资料。

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

# view()    转换维度
# reshape() 转换维度
# permute() 坐标系变换
# squeeze()/unsqueeze() 降维/升维
# expand()   扩张张量
# narraw()   缩小张量
# resize_()  重设尺寸
# repeat(), unfold() 重复张量
# cat(), stack()     拼接张量

文章链接：https://blog.youkuaiyun.com/x_yan033/article/details/104965077

有详细总结tensor维度转换函数的使用方法及常见场景。