PyTorch的 torch.unsqueeze() 和 torch.squeeze()方法详解

本文介绍了PyTorch库中的torch.unsqueeze()和torch.squeeze()函数,分别用于在指定维度增加1维和移除单维度,以便调整张量的形状,常用于处理批处理和优化内存使用。
摘要由CSDN通过智能技术生成

torch.unsqueeze()

  • 定义: torch.unsqueeze(input, dim) 返回一个新的张量,该张量为指定位置dim增加了一个大小为1的维度。

  • 参数:

    • input (Tensor): 输入张量。
    • dim (int): 插入维度的索引。
  • 返回值: Tensor。一个新的张量,它与输入张量input有着几乎相同的数据,但是增加了一个维度。

  • 用途: 增加张量的一个维度,常在希望增加一个批处理维度时使用。

  • 代码示例:

    import torch
    
    x = torch.tensor([1, 2, 3, 4])
    y = torch.unsqueeze(x, 0)  # 结果为:[[1, 2, 3, 4]]
    

    torch.squeeze()

  • 定义: torch.squeeze(input, dim=None) 返回一个新的张量,它会移除input中所有指定维度dim的1维。

  • 参数:

    • input (Tensor): 输入张量。
    • dim (int, 可选): 要压缩的维度。如果未指定,则移除所有单维度。
  • 返回值: Tensor。一个新的张量,它与输入张量input有着相同的数据,但是移除了所有指定的单维度。

  • 用途: 减少张量的维度,常用于移除批处理维度或任何大小为1的维度。

  • 代码示例:

    import torch
    
    x = torch.tensor([[1, 2, 3, 4]])
    y = torch.squeeze(x, 0)  # 结果为:[1, 2, 3, 4]
    

         注意:在实际代码中,返回的张量通常会共享输入张量的数据,除非必须复制(比如要求返回一个不同内存的张量)。

  • 8
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

实名吃香菜

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值