x.squeeze()、x.unsqueeze()用法
x.squeeze(dim)
用途:进行维度压缩,去掉tensor中维数为1的维度
参数设置:如果设置dim=a,就是去掉指定维度中维数为1的
示例:
import torch
x = torch.tensor([[[1],[2]],[[3],[4]]])
print('x:',x)
x1 = x.squeeze()
print('x1:',x1)
x2 = x.squeeze(2)
print('x2:',x2)
输出:
x: tensor([[[1],[2]],[[3],[4]]])
x1: tensor([[1, 2],[3, 4]])
x2: tensor([[1, 2],[3, 4]])Process finished with exit code 0
x.unsqueeze(dim=a)
用途:进行维度扩充,在指定位置加上维数为1的维度
参数设置:如果设置dim=a,就是在维度为a的位置进行扩充
示例:
import torch
x = torch.tensor([1,2,3,4])
print(x)
x1 = x.unsqueeze(dim=0)
print(x1)
x2 = x.unsqueeze(dim=1)
print(x2)y = torch.tensor([[1,2,3,4],[9,8,7,6]])
print(y)
y1 = y.unsqueeze(dim=0)
print(y1)
y2 = y.unsqueeze(dim=1)
print(y2)
输出:
x: tensor([1, 2, 3, 4])
x1: tensor([[1, 2, 3, 4]])
x2: tensor([[1],[2],[3],[4]])
y: tensor([[1, 2, 3, 4],[9, 8, 7, 6]])
y1: tensor([[[1, 2, 3, 4],[9, 8, 7, 6]]])
y2: tensor([[[1, 2, 3, 4]],[[9, 8, 7, 6]]])Process finished with exit code 0
x.squeeze()、x.unsqueeze()用法相关推荐
- pytorch中squeeze()与unsqueeze()的用法
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档 文章目录 一.pytorch中的数据维度 二.squeeze()与unsqueeze()用法 1.squeeze() 2.unsque ...
- pytorch中的squeeze和unsqueeze的用法小结
文章目录 1. 理论简介 2. 举例 2.1 squeeze (1)x.squeeze() (2)x.squeeze(n) (3)torch.squeeze(x) 2.2 unsqueeze (1)x ...
- python中squeeze函数_详解pytorch中squeeze()和unsqueeze()函数介绍
squeeze的用法主要就是对数据的维度进行压缩或者解压. 先看torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一行三列(1,3)的 ...
- squeeze()和unsqueeze()
torch.squeeze() 两种用法:一种直接a.squeeze():另一种torch.squeeze(a) 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一 ...
- Pytorch view()、squeeze()、unsqueeze()、torch.max()
本篇博客主要向大家介绍Pytorch中view().squeeze().unsqueeze().torch.max()函数,这些函数虽然简单,但是在 神经网络编程总却经常用到,希望大家看了这篇博文能够 ...
- torch.squeeze()和unsqueeze()
torch.squeeze()和unsqueeze() unsqueeze() 函数功能:与squeeze()函数功能相反,用于添加维度. queeze() 函数功能:去除size为1的维度,包括行和 ...
- PyTorch中为什么需要使用squeeze()和unsqueeze()操作?
1 问题描述 今天在学习FaceBoxes~ 在看到模型推理部分的代码时,总会出现squeeze()和unsqueeze()操作,从网上的资料上来看,就是升维和降维的操作,但是为什么需要进行这样的操作 ...
- 【Pytorch】tensor类型数据.squeeze()和.unsqueeze()函数的简明教程(一看就会)
文章目录 1 squeeze 1.1 1.2 1.3 执行操作后需要写回 2 unsqueeze pytorch系列代码中常见的两个函数squeeze()和unsqueeze() 1 squeeze ...
- PyTorch中squeeze()和unsqueeze()详解
pytorch中squeeze()和unsqueeze()作用 squeeze() squeeze() 用于在张量的指定维度插入新的维度 (为1) 得到维度提升的张量. unsqueeze() uns ...
- Pytorch squeeze() unsqueeze() 用法
简介 torch.squeeze(input, dim=None, out=None):对数据的维度进行压缩,去掉维数为1的的维度. squeeze函数功能:去除size为1的维度,包括行和列.当维度 ...
最新文章
- 一个fork短码的扩展版本
- 在制造业大厂当程序员是什么感受?
- mybatis collection用法_Mybatis中强大的resultMap
- Linux+Apache2.4+PHP5.6+MySQL5.6源码安装步骤
- 微信小程序 App()方法与getApp()方法
- 内存四区(代码区 静态区 栈区 堆区)
- CSS3景深-perspective
- Java集合之Hashtable源码分析
- java 减少内存_java – 减少内存流失的方法
- CSDN、博文视点名家讲坛之夏昕、林信良谈Spring框架
- java读取 文件_Java读取文件的简单实现方法
- 加载类型库/DLL 时出错。 (Exception from HRESULT: 0x80029C4A (TYPE_E_CANTLOADLIBRARY)
- 单阶段目标检测算法YOLOv3
- 浅谈Ubuntu 18.04.1 LTS x86_64安装,美化配置及常用软件安装配置的历程
- 19c 单实例打补丁
- 全连接神经网络(FCNN)
- wi8ndows无法加载,Win8.1系统更新Flash插件后无法自动加载插件怎么办
- 你一定想不到,动物和AI之间的联系如此深刻!
- java设置右击菜单_windows7中右键菜单添加powershell
- 05.视频播放器内核切换封装
热门文章
- Unity3D中的空间与运动
- 各种内容转载以及PS
- Oracle系列十二:视图、记录、同义词、序列
- 供水管网管道泄漏监测系统定位技术分析方案
- JVM内功心法-GC垃圾回收之GC垃圾回收过程
- 港版AI chat如何制作?
- 当下,产业园区发展面临的十大问题
- 上线 B 站,钢铁侠出镜 AI 科普纪录片!
- asked CMake to find a package configuration file provided by MRPT, but CMake did not find one.
- java语言与其他语言的区别是_Java语言特点有哪些?Java区别其他语言特点详解