squeeze 、unsequeeze

两者都是作用于tensor。

记住单词：squeeze是挤压、塞入的意思

unsqueeze(num)是增加维度，num代表增加维度所在的位置。

squeeze(num)是减少维度，num代表的是减少维度所在的位置，所减少的维度只能是1维，也就是不存在数据的维度。

import torch
a=torch.arange(0,6)
a=a.view(2,3)
print(a.shape)#torch.Size([2,3])b=a.unsqueeze(1)
print(b.shape)#torch.Size([2,1,3])
c=a.squeeze(0)
print(c.shape)#torch.Size([2,3])
d=a.unsqueeze(2)
print(d.shape)#torch.Size([2,3,1])
e=d.squeeze(2)
print(e.shape)#torch.Size([2,3])

version.shape可查看tensor的shape。

squeeze 、unsequeeze相关推荐

Pytorch view()、squeeze()、unsqueeze()、torch.max()
本篇博客主要向大家介绍Pytorch中view().squeeze().unsqueeze().torch.max()函数,这些函数虽然简单,但是在神经网络编程总却经常用到,希望大家看了这篇博文能够 ...
numpy ravel、flatten、squeeze、reshape
numpy:手册一.ravel.flatten.squeeze.reshape ravel(): 不会产生原来数据的副本 flatten():返回源数据副本 squeeze():只能对维度为1的维度 ...
permute、transpose、view、reshape、unsequeeze与flatten
Pytorch中view, transpose, permute等方法的区别关于张量的Flatten.Reshape和Squeeze的解释 transpose 只能交换两个维度,参数是(dim1,d ...
Pytorch教程（十三）：reshape、squeeze、flatten
reshape 首先我们创建一个tensor,shape:3×43 \times 43×4,是一个两个轴的二阶张量.第一个轴的长度是3,第二个轴的长度是4,第一个轴的元素是数组,第二个轴的元素是数字 ...
【增减维度】numpy和torch中的squeeze、unsqueeze理解
文章目录 1 为何要增减维度 2 numpy中的squeeze 函数 3 torch中的squeeze 函数 4 torch中的unsqueeze 函数 1 为何要增减维度神经网络conv2d的输入 ...
x.squeeze()、x.unsqueeze()用法
x.squeeze(dim) 用途:进行维度压缩,去掉tensor中维数为1的维度参数设置:如果设置dim=a,就是去掉指定维度中维数为1的示例: import torch x = torch.t ...
PyTorch常用的张量创建、变形及运算总结（速查表）
PyTorch张量常用的创建.变形及数学运算总结目录 PyTorch张量常用的创建.变形及数学运算总结 1. 张量(tensor)的创建 1.1 torch.Tensor()与torch.tenso ...
Android Pie提供了自适应供电、神经网络API 1.1等新特性
Google发布了Android Oreo的后继版本Android Pie.Android Pie提供了"刘海"屏显示支持(Display cutout).神经网络API 1.1. ...
树莓派AI视觉云台——2、树莓派系统镜像的下载和烧写
一．树莓派启动方式及支持的系统树莓派开发板没有配置板载FLASH,因为它支持SD卡启动,所有我们需要下载相应镜像,并将其烧写在SD上,启动系统即可.(这个镜像里包含了我们通常所说的bootloade ...

squeeze 、unsequeeze

squeeze 、unsequeeze相关推荐

最新文章

热门文章