• pytorch的unsqueeze与squeeze函数


    torch.squeeze()

    这个函数主要对数据的维度进行压缩,去掉维数为1的的维度。
    比如是一行或者一列这种,一个一行三列(1,3)的数去掉第一个维数为一的维度之后就变成(3)行。
    squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。
    a.squeeze(dim) 就是去掉a中指定的维数为一的维度。
    还有一种形式就是b=torch.squeeze(a,dim) a中去掉指定的定的维数为一的维度。

    torch.unsqueeze()

    这个函数主要是对数据维度进行扩充,给指定位置加上维数为一的维度。
    比如原本有个三行的数据(3),在0的位置加了一维就变成一行三列(1,3)。
    a.squeeze(dim) 就是在a中指定位置dim加上一个维数为1的维度。
    还有一种形式就是b=torch.squeeze(a,dim) a就是在a中指定位置dim加上一个维数为1的维度

    实例

  • 相关阅读:
    OpenJudge 6042 雇佣兵
    POJ 1741 树上 点的 分治
    Codevs 1695 Windows2013
    复制书稿
    乘积最大
    编辑距离问题
    石子合并
    最大正方形子矩阵
    选菜
    混合背包
  • 原文地址:https://www.cnblogs.com/Higgerw/p/14041406.html
Copyright © 2020-2023  润新知