首页 > 其他分享 >【d2l】【常见函数】【19】 torch.squeeze( ), torch.unsqueeze( )

【d2l】【常见函数】【19】 torch.squeeze( ), torch.unsqueeze( )

时间:2023-07-29 18:12:27浏览次数:45  
标签:unsqueeze 19 docs torch 维度 squeeze

torch.squeeze()

压缩大小为1的维度
参考:https://pytorch.org/docs/stable/generated/torch.squeeze.html
image
image
image

torch.unsqueeze()

在指定的维度处增加一个维度,大小为1
参考:https://pytorch.org/docs/stable/generated/torch.unsqueeze.html
image
image

标签:unsqueeze,19,docs,torch,维度,squeeze
From: https://www.cnblogs.com/lycnight/p/17590225.html

相关文章

  • 【d2l】【常见函数】【17】 torch.rand( (size) )
    返回一个指定size的张量,元素是0~1之间的随机数参考:https://pytorch.org/docs/stable/generated/torch.rand.html......
  • 【d2l】【常见函数】【16】 torch.repeat_interleave( )
    参考:https://pytorch.org/docs/stable/generated/torch.repeat_interleave.html......
  • 2019年湖南省对口高考真题
    一、选择题1、已知"intw=1,x=2,y=3,z=4;",则表达式"w>x?w:z>y?z:x"的值是__________。A.1               B.2               C.3               D.42、若有定义语句"inta[3][5];",按内存中的数据存放顺序,a数组的第10个元素是_____......
  • 最完美WIN10_Pro_22H2.19045.3271软件选装纯净版VIP51.1
    【系统简介】=============================================================1.本次更新母盘来自UUP_WIN10_PRO_22H2.19045.3271。进一步精简优化调整。2.只为呈现最好的作品,手工精简优化部分较多。3.OS版本号为19045.3271。个别要求高的就下MSDN吧,里面啥功能都有。4.集成《DrvCeo......
  • 多层感知机——pytorch版
    importtorchfromtorchimportnnfromd2limporttorchasd2lbatch_size=256train_iter,test_iter=d2l.load_data_fashion_mnist(batch_size)num_inputs,num_outputs,num_hiddens=784,10,256w1=nn.Parameter(torch.randn(num_inputs,num_hiddens,re......
  • 线性回归模型实现——pytorch版
    importrandomimporttorchfromd2limporttorchasd2ldefsynthetic_data(w,b,num_examples):"""生成y=Xw+b+噪声"""x=torch.normal(0,1,(num_examples,len(w)))#01正态分布,num_examples个样本,len(w)列print('len是:'......
  • torch.all()
    torch.all()其中的所有项都为True,返回True,反之,返回False>>>importtorch>>>a=torch.tensor([[1,2],[3,4]])>>>b=torch.tensor([[1,2],[3,4]])>>>aOUT:tensor([[1,2],[3,4]])>>>torch.eq(a,b)OU......
  • 苹果mac m1,m2芯片安装 pytorch和tensorflow的GPU版本
    一、下载M芯片的anaconda,并安装 二、安装GPU版本的pytorch1.安装Xcodexcode-select--install2.创建环境condacreate-ntorch-gpupython=3.11condaactivatetorch-gpu 3.打开pytorch官网复制命令,注意:在macm上,device是’mps’而不是’cuda’,mac的MPS......
  • PyTorch 1.4 中文文档校对活动正式启动 | ApacheCN
    一如既往,PyTorch1.4中文文档校对活动启动了!认领须知请您勇敢地去翻译和改进翻译。虽然我们追求卓越,但我们并不要求您做到十全十美,因此请不要担心因为翻译上犯错——在大部分情况下,我们的服务器已经记录所有的翻译,因此您不必担心会因为您的失误遭到无法挽回的破坏。(改编自维基百......
  • ApacheCN 活动汇总 2019.8.16
    公告欢迎大家在我们平台上投放广告。如果你希望在我们的专栏、文档或邮件中投放广告,请准备好各种尺寸的图片和专属链接,我们组织了一个开源互助平台,方便开源组织和大V互相认识,互相帮助,整合资源。请回复这个帖子并注明组织/个人信息来申请加入。请回复这个帖子来推荐希望翻译的内容......