【学习笔记】pytorch中squeeze()和unsqueeze()函数介绍

一、unsqueeze()函数

1. 首先初始化一个a

可以看出a的维度为（2，3）

2. 在第二维增加一个维度，使其维度变为（2，1，3）

可以看出a的维度已经变为（2，1，3）了，同样如果需要在倒数第二个维度上增加一个维度，那么使用b.unsqueeze(-2)

二、squeeze()函数介绍

1. 首先得到一个维度为（1，2，3）的tensor（张量）

由图中可以看出c的维度为（1，2，3）

2.下面使用squeeze()函数将第一维去掉

可见，维度已经变为（2，3）

3.另外

可以看出维度并没有变化，仍然为（1，2，3），这是因为只有维度为1时才会去掉。

【学习笔记】pytorch中squeeze()和unsqueeze()函数介绍相关推荐

python中squeeze函数_详解pytorch中squeeze()和unsqueeze()函数介绍
squeeze的用法主要就是对数据的维度进行压缩或者解压. 先看torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一行三列(1,3)的 ...
pytorch 中 squeeze 和unsqueeze函数
1. torch.squeeze() 函数 : 作用:移除指定或所有维数为1的维度,从而得到维度减少的张量解释一下: x=torch.zeros(5,1,1,1)print(x)'输出' tenso ...
PyTorch中squeeze()和unsqueeze()详解
pytorch中squeeze()和unsqueeze()作用 squeeze() squeeze() 用于在张量的指定维度插入新的维度 (为1) 得到维度提升的张量. unsqueeze() uns ...
[PyTorch] 深度学习框架PyTorch中的概念和函数
Pytorch的概念 Pytorch最重要的概念是tensor,意为"张量". Variable是能够构建计算图的 tensor(对 tensor 的封装).借用Variable才 ...
pytorch中squeeze()与unsqueeze()的用法
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档文章目录一.pytorch中的数据维度二.squeeze()与unsqueeze()用法 1.squeeze() 2.unsque ...
pytorch中tensor的unsqueeze()函数和squeeze()函数的用处
unsqueeze()用于增加一个维度. 先假设有如下一维的Tensor. a=torch.Tensor([1,2]) print(a.shape) 假设我们现在有一个2*2的矩阵b,要与a相乘,最规 ...
Pytorch：squeeze()和unsqueeze()函数
squeeze(): 维度压缩,去掉维数为1的的维度 torch.squeeze(input, dim=None, out=None) 将输入张量形状中的1 去除并返回. 如果输入是形如(2×1×2× ...
Linux学习笔记-Makefile中的变量及函数
Makefile中的变量此处的变量不是C/C++,Java等中的变量(认为他是一个字符串就阔以了) 一般变量名用大写字母来写,如下所属: SUBDIR = src xml SUBDIR += osa ...
r语言c函数怎么用,R语言学习笔记——C#中如何使用R语言setwd()函数
在R语言编译器中,设置当前工作文件夹可以用setwd()函数. > setwd("e://桌面//") > setwd("e:\桌面\") > ...

【学习笔记】pytorch中squeeze()和unsqueeze()函数介绍

【学习笔记】pytorch中squeeze()和unsqueeze()函数介绍相关推荐

最新文章

热门文章