- 博客(2)
- 收藏
- 关注
原创 pytorch中的torch.squeeze()函数
如果 input 的形状为 (A×1×B),squeeze(input, 0)后,返回的tensor不变;squeeze(input, 1)后,返回的tensor将被压缩为 (A×B)上次tensor([16,1,1,1]),需要转换成tensor([16,1]),,就写 torch.squeeze(input, dim=2,3),转换成功。squeeze()函数的功能是维度压缩。如果 input 的形状为 (A×1×B×C×1×D),那么返回的tensor的形状则为 (A×B×C×D)
2024-03-27 11:51:45
217
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人