脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → tensor.squeeze函数和tensor.unsqueeze函数

tensor.squeeze函数和tensor.unsqueeze函数的使用详解

 更新时间：2023年03月09日 08:58:48   作者：不想悲伤到天明  
本文主要介绍了tensor.squeeze函数和tensor.unsqueeze函数的使用详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

tensor.squeeze() 和 tensor.unsqueeze() 是 PyTorch 中用于改变 tensor 形状的两个函数，它们的作用如下：

  • tensor.squeeze(dim=None, *, out=None) : 压缩 tensor 中尺寸为 1 的维度，并返回新的 tensor。可以指定要压缩的维度（默认为所有尺寸为 1 的维度均压缩）。
  • tensor.unsqueeze(dim, *, out=None) : 在指定的位置插入一个新维度，并返回新的 tensor。dim 参数表示新插入的维度在哪个位置（从 0 开始），可以是负数，表示倒数第几个维度。
  • squeeze 是压缩维度，unsqueeze是增加维度.

下面给出例子来说明它们的使用。

tensor.squeeze()

import torch
 
# 创建一个形状为 (1, 3, 1, 2) 的 tensor
x = torch.randn(1, 3, 1, 2)
print(x.shape)  # torch.Size([1, 3, 1, 2])
 
# 压缩尺寸为 1 的维度
y = x.squeeze()
print(y.shape)  # torch.Size([3, 2])
 
# 指定要压缩的维度
y = x.squeeze(dim=0)
print(y.shape)  # torch.Size([3, 1, 2])

在上面的例子中，我们创建了一个形状为 (1, 3, 1, 2) 的 tensor，然后使用 squeeze() 函数压缩了尺寸为 1 的维度。在第二个 squeeze() 调用中，我们指定了要压缩的维度为 0，也就是第一个维度，因此第一个维度的大小被压缩为 1，变成了形状为 (3, 1, 2) 的 tensor。

tensor.unsqueeze()

import torch
 
# 创建一个形状为 (3, 2) 的 tensor
x = torch.randn(3, 2)
print(x.shape)  # torch.Size([3, 2])
 
# 在维度 0 上插入新维度
y = x.unsqueeze(dim=0)
print(y.shape)  # torch.Size([1, 3, 2])
 
# 在维度 1 上插入新维度
y = x.unsqueeze(dim=1)
print(y.shape)  # torch.Size([3, 1, 2])
 
# 在倒数第二个维度上插入新维度
y = x.unsqueeze(dim=-2)
print(y.shape)  # torch.Size([3, 1, 2])

在上面的例子中，我们创建了一个形状为 (3, 2) 的 tensor，然后使用 unsqueeze() 函数在不同的位置插入了新维度。在第一个 unsqueeze() 调用中，我们在维度 0 上插入了新维度，因此新的 tensor 形状为 (1, 3, 2)。在第二个和第三个 unsqueeze() 调用中，我们分别在维度 1 和倒数第二个维度上插入了新维度，分别得到了形状为 (3, 1, 2) 和 (3, 2, 1) 的 tensor。

到此这篇关于tensor.squeeze函数和tensor.unsqueeze函数的使用详解的文章就介绍到这了,更多相关tensor.squeeze函数和tensor.unsqueeze函数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

相关文章

  • Python numpy中矩阵的基本用法汇总

    Python numpy中矩阵的基本用法汇总

    这篇文章主要给大家介绍了关于Python numpy中矩阵的基本用法的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
    2019-02-02
  • python聚类算法解决方案（rest接口/mpp数据库/json数据/下载图片及数据）

    python聚类算法解决方案（rest接口/mpp数据库/json数据/下载图片及数据）

    这篇文章主要介绍了python聚类算法解决方案（rest接口/mpp数据库/json数据/下载图片及数据）,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • 浅谈Python魔法方法

    浅谈Python魔法方法

    今天给大家带来的是关于Python的相关知识,文章围绕着Python魔法方法展开,文中有非常详细的介绍及代码示例,需要的朋友可以参考下
    2021-06-06
  • python相似模块用例

    python相似模块用例

    这篇文章主要为大家详细介绍了python相似模块用例，感兴趣的小伙伴们可以参考一下
    2016-03-03
  • python flask解析json数据不完整的解决方法

    python flask解析json数据不完整的解决方法

    这篇文章主要介绍了python flask解析json数据不完整的解决方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2019-05-05
  • python实现FFT快速傅立叶变换算法案例

    python实现FFT快速傅立叶变换算法案例

    FFT（快速傅里叶变换）是计算DFT及其逆变换的一种算法,其基本思想是利用DFT的对称性和周期性,通过分而治之的策略将DFT分解为更小的DFT,从而降低计算复杂度,FFT的算法步骤包括选择分解、重新排序、蝶形运算和逐层计算,在Python中
    2024-10-10
  • python动态加载变量示例分享

    python动态加载变量示例分享

    这篇文章主要介绍了python动态加载变量示例,需要的朋友可以参考下
    2014-02-02
  • Python中的defaultdict与__missing__()使用介绍

    Python中的defaultdict与__missing__()使用介绍

    下面这篇文章主要给大家介绍了关于Python中defaultdict使用的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用python具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。
    2018-02-02
  • Python Type Hints 学习之从入门到实践

    Python Type Hints 学习之从入门到实践

    Type Hints（类型注解）进一步强化了Python是一门强类型语言的特性，它在 Python3.5 中第一次被引入。使用Type Hints可以让我们编写出带有类型的Python代码，本文将详细介绍一下Type Hints，感兴趣的小伙伴可以关注一下
    2021-11-11
  • 用python写PDF转换器的实现

    用python写PDF转换器的实现

    这篇文章主要介绍了用python写PDF转换器的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-10-10

最新评论