欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

pytorch tensor张量维度转换

程序员文章站 2022-06-11 22:08:55
...
# view()    转换维度
# reshape() 转换维度
# permute() 坐标系变换
# squeeze()/unsqueeze() 降维/升维
# expand()   扩张张量
# narraw()   缩小张量
# resize_()  重设尺寸
# repeat(), unfold() 重复张量
# cat(), stack()     拼接张量

文章链接:https://blog.csdn.net/x_yan033/article/details/104965077

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

相关标签: PyTorch