在pytorch中我们经常会遇到维度变换的情况,接下来介绍常见3中维度变换的函数。
1 view 和 reshape两个函数功能和用法都一样
输出:
1 unsqueeze
具体应用场合
2 squeeze
expend 只是增加了理解方式,并没有拷贝数据(推荐)
repeat 拷贝了数据

在pytorch中我们经常会遇到维度变换的情况,接下来介绍常见3中维度变换的函数。
1 view 和 reshape两个函数功能和用法都一样
输出:
1 unsqueeze
具体应用场合
2 squeeze
expend 只是增加了理解方式,并没有拷贝数据(推荐)
repeat 拷贝了数据