pytorch中squeeze()和unsqueeze()作用

squeeze()

squeeze() 用于在张量的指定维度插入新的维度 (为1) 得到维度提升的张量。

unsqueeze()

unsqueeze() 用于移除指定或者所有维度大小为1的维度，从而得到维度减小的新张量。

演示代码：

import torcha=torch.arange(0,6)
b=a.view(2,3) # 维度：2*3
print("b",b)c=b.unsqueeze(0) # 维度：1*2*3
print("c",c)d=c.unsqueeze(2) # 维度：1*2*1*3
print("d",d)e=d.squeeze(0) # 维度：2*1*3
print("e",e)f=e.squeeze(0) # 维度：2*1*3
print("f",f)

结果显示：

b tensor([[0, 1, 2],[3, 4, 5]])
c tensor([[[0, 1, 2],[3, 4, 5]]])
d tensor([[[[0, 1, 2]],[[3, 4, 5]]]])
e tensor([[[0, 1, 2]],[[3, 4, 5]]])
f tensor([[[0, 1, 2]],[[3, 4, 5]]])

PyTorch中squeeze()和unsqueeze()详解相关推荐

python中squeeze函数_详解pytorch中squeeze()和unsqueeze()函数介绍
squeeze的用法主要就是对数据的维度进行压缩或者解压. 先看torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一行三列(1,3)的 ...
pytorch中squeeze()与unsqueeze()的用法
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档文章目录一.pytorch中的数据维度二.squeeze()与unsqueeze()用法 1.squeeze() 2.unsque ...
PyTorch中计算KL散度详解
PyTorch计算KL散度详解最近在进行方法设计时,需要度量分布之间的差异,由于样本间分布具有相似性,首先想到了便于实现的KL-Divergence,使用PyTorch中的内置方法时,踩了不少坑,在 ...
pytorch中的卷积操作详解
首先说下pytorch中的Tensor通道排列顺序是:[batch, channel, height, width] 我们常用的卷积(Conv2d)在pytorch中对应的函数是: torch.nn. ...
PyTorch中的matmul函数详解
PyTorch中的两个张量的乘法可以分为两种: 两个张量对应的元素相乘(element-wise),在PyTorch中可以通过torch.mul函数(或者∗*∗运算符)实现两个张量矩阵相乘(Matr ...
PyTorch中torch.norm函数详解
torch.norm() 是 PyTorch 中的一个函数,用于计算输入张量沿指定维度的范数.具体而言,当给定一个输入张量 x 和一个整数 p 时,torch.norm(x, p) 将返回输入张量 x ...
PyTorch中的squeeze()和unsqueeze()详解与应用案例
文章目录 1.torch.squeeze 2.torch.unsqueeze 3.给一个使用上述两个函数,并进行一次卷积的例子附上官网地址: https://pytorch.org/docs/sta ...
【学习笔记】pytorch中squeeze()和unsqueeze()函数介绍
一.unsqueeze()函数 1. 首先初始化一个a 可以看出a的维度为(2,3) 2. 在第二维增加一个维度,使其维度变为(2,1,3) 可以看出a的维度已经变为(2,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()

unsqueeze()

演示代码：

PyTorch中squeeze()和unsqueeze()详解相关推荐

最新文章

热门文章