文章目录 一、unsqueeze() 方法 二、实例 参考链接 一、unsqueeze() 方法 unsqueeze() 方法起升维的作用，参数表示在哪个地方加一个维度。

二、实例 import torch input=torch.arange(0,6) print(input) print(input.shape)

print(input.unsqueeze(0))    # 在0维处加一维 print(input.unsqueeze(0).shape)

print(input.unsqueeze(1))    # 在1维处加一维 print(input.unsqueeze(1).shape)   输出结果：

tensor([0, 1, 2, 3, 4, 5]) torch.Size([6]) tensor([[0, 1, 2, 3, 4, 5]]) torch.Size([1, 6])  # 在0维处加了一维 tensor([[0],         [1],         [2],         [3],         [4],         [5]]) torch.Size([6, 1])  # 在1维处加了一维   参考链接 pytorch中unsqueeze()函数理解 ———————————————— 版权声明：本文为CSDN博主「想变厉害的大白菜」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。 原文链接：https://blog.csdn.net/weixin_44211968/article/details/128223412