Pytorch squeeze()的简单用法

squeeze的作用就是对tensor变量进行维度压缩，去除维数为1的的维度。
☆例如，一个3×2×1×2×1的tensor，squeeze()之后便成了3×2×2。存储的数据并没有发生变化，但是去除了“多余”的维度信息：

import torchx = torch.rand(3, 2, 1, 2, 1)
print(x.size())
x = x.squeeze()
print(x.size())

输出：

torch.Size([3, 2, 1, 2, 1])
torch.Size([3, 2, 2])

☆如果原tensor没有维度为1，则不会进行任何操作：

import torchx = torch.rand(2, 2)
print(x)
x = x.squeeze()
print(x)

输出：

tensor([[0.6088, 0.5360],[0.4375, 0.0350]])
tensor([[0.6088, 0.5360],[0.4375, 0.0350]])

☆此外，也可以指定尝试将哪一个维度进行压缩。如果被指定的维度其维数为1，则压缩，反之不对该维度操作：

import torchx = torch.rand(3, 2, 1, 2, 1)
print(x.size())
x = x.squeeze(0)
print(x.size())
x = x.squeeze(2)
print(x.size())

输出：

torch.Size([3, 2, 1, 2, 1])
torch.Size([3, 2, 1, 2, 1])
torch.Size([3, 2, 2, 1])

☆torch中的squeeze方法与上面提到的Tensor对象中的方法作用相同，语法有细微区别：

import torch
x = torch.rand(3, 2, 1, 2, 1)
x = torch.squeeze(x) # 同x.squeeze()
print(x.size())
x = torch.squeeze(x, 2) # 同x.squeeze(2)
print(x.size())

Pytorch squeeze()的简单用法相关推荐

Pytorch permute()的简单用法
作用:调换Tensor中各维度的顺序输入:数组,用来表示调换后的各维度顺序.一开始没修改的顺序其实就是[0,1,2-] 例子:有一个6×7×8×9的Tensor,我们想将其第2维(8)和第三维(9) ...
反编译工具jad简单用法
反编译工具jad简单用法下载地址: [url]http://58.251.57.206/down1?cid=B99584EFA6154A13E5C0B273C3876BD4CC8CE672& ...
QCustomPlot的简单用法总结
QCustomPlot的简单用法总结第一部分:QCustomPlot的下载与安装第二部分:QCustomPlot在VS2013+QT下的使用 QCustomPlot的简单用法总结写在前面, ...
python matplotlib 简单用法
python matplotlib 简单用法具体内容请参考官网代码 import matplotlib.pyplot as plt import numpy as np # 支持中文 plt.rc ...
Windump网络命令的简单用法
Windump网络命令的简单用法大家都知道,unix系统下有个tcpdump的抓包工具,非常好用,是做troubleshooting的好帮手.其实在windows下也有一个类似的工作,叫windum ...
Android TabLayout(选项卡布局)简单用法实例分析
本文实例讲述了Android TabLayout(选项卡布局)简单用法.分享给大家供大家参考,具体如下: 我们在应用viewpager的时候,经常会使用TabPageIndicator来与其配合.达到 ...
shell expect的简单用法
为什么需要expect? 我们通过Shell可以实现简单的控制流功能,如:循环.判断等.但是对于需要交互的场合则必须通过人工来干预,有时候我们可能会需要实现和交互程序如 telnet服务器等进 ...
Shellz中awk的简单用法
其实shell脚本的功能常常被低估.在实际应用中awk sed 等用法可以为shell提供更为强大的功能.下面我们将一下awk调用的简单方法进行了总结.方便同学们学习: awk的简单用法: 第一种调用 ...
python装饰器实例-Python装饰器原理与简单用法实例分析
本文实例讲述了Python装饰器原理与简单用法.分享给大家供大家参考,具体如下: 今天整理装饰器,内嵌的装饰器.让装饰器带参数等多种形式,非常复杂,让人头疼不已.但是突然间发现了装饰器的奥秘,原来如此 ...

Pytorch squeeze()的简单用法

Pytorch squeeze()的简单用法相关推荐

最新文章

热门文章