- 博客(2)
- 收藏
- 关注
原创 pytorch中的squeeze和unsqueeze的用法
squeeze :对tensor变量进行维度压缩,去除维数为1的维度。x.squeeze(n):在维度n上,进行维度压缩,如果是1维的就压缩,不是就不压。
2024-01-24 17:51:51 330 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
squeeze :对tensor变量进行维度压缩,去除维数为1的维度。x.squeeze(n):在维度n上,进行维度压缩,如果是1维的就压缩,不是就不压。
2024-01-24 17:51:51 330 1
TA创建的收藏夹 TA关注的收藏夹
TA关注的人