x = x.view(x.size(0), -1) 的理解_x = x.view(x.size(0),-1)-程序员宅基地

技术标签: python  机器学习  pytorch  

之前对于pytorch的网络编程学习都是大致理解每一层的概念,有些语法语句没有从原理上弄清楚,就比如标题的x = x.view(x.size(0), -1)  。

这句话一般出现在model类的forward函数中,具体位置一般都是在调用分类器之前。分类器是一个简单的nn.Linear()结构,输入输出都是维度为一的值,x = x.view(x.size(0), -1)  这句话的出现就是为了将前面多维度的tensor展平成一维。下面是个简单的例子,我将会根据例子来对该语句进行解析。

class NET(nn.Module):
    def __init__(self,batch_size):
        super(NET,self).__init__()
        self.conv = nn.Conv2d(outchannels=3,in_channels=64,kernel_size=3,stride=1)
        self.fc = nn.Linear(64*batch_size,10)

    def forward(self,x):
        x = self.conv(x)
        x = x.view(x.size(0), -1)  
        out = self.fc(x)

上面是个简单的网络结构,包含一个卷积层和一个分类层。forward()函数中,input首先经过卷积层,此时的输出x是包含batchsize维度为4的tensor,即(batchsize,channels,x,y),x.size(0)指batchsize的值。 x = x.view(x.size(0), -1)简化x = x.view(batchsize, -1)。

view()函数的功能根reshape类似,用来转换size大小。x = x.view(batchsize, -1)中batchsize指转换后有几行,而-1指在不告诉函数有多少列的情况下,根据原tensor数据和batchsize自动分配列数。

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/whut_ldz/article/details/78882532

智能推荐

Vlog简介-程序员宅基地

文章浏览阅读995次。https://www.ifanr.com/1138470转载于:https://www.cnblogs.com/pengwang52/p/10683069.html_校园vlog简介怎么写

python稳健回归_【Stata教程】如何用stata做稳健回归-程序员宅基地

文章浏览阅读1.2k次。“社会科学中的数据可视化”第411篇推送导言大量的线性回归模型是基于最小二乘法实现的,但其仍存在一些局限性。比如说,样本点出现许多异常点时,传统的最小二乘法将不再适用,此时则可以使用稳健回归(robust regression)代替最小二乘法。操作下面的稳健回归使用的是犯罪数据,该数据来自Alan Agresti和Barbara Finlay的《社会科学统计方法》。变量包括美国各州编号(sid)、..._margins 贫困

爬虫.requests.exceptions.ConnectionErro-程序员宅基地

文章浏览阅读219次。requests.exceptions.ConnectionError: HTTPConnectionPool(host='jy-qj.com.cn', port=80): Max retries exceeded with url: / (Caused by NewConnectionError('<requests.packages.urllib3.connection.HTTPConn..._requests.exceptions.connectionerror: errno1104 getaddrinfo failed

[C++]欧几里得辗转相除求最大公约数,练习_欧几里得算法c++练习题-程序员宅基地

文章浏览阅读1.1k次。编程实现求解最大公约数的欧几里德算法,用户输入两个任意正整数,程序输出他们的最大公约数。算法如下:拆解步骤如下:步骤1: 如果p < q,则交换p和q。步骤2: 令r是p / q 的余数。步骤3: 如果r = 0,则令g = q并终止;否则令p = q, q = r并转向步骤2#include<iostream>#include<stdio.h>//编程实现求解最大公约数的欧几里德算法,用户输入两..._欧几里得算法c++练习题

ViewPager的notifyDataSetChanged()没有效果?来从源码上解决这个问题_viewpager notifydatasetchanged-程序员宅基地

文章浏览阅读841次。前言最近发现自己有很多颇为基础的内容“不会写”了,就比如今天写的内容:ViewPager。最近有小伙伴,在后台私信一些技术细节,大家真的好勤奋~~因为工作的原因,有些私信回复的不是很及时,多多包涵。996伤不起啊!正文平时我们很容易遇到这样的需求:页面底部很多Tab,可以点击或者活动切换不同的页面…估计话还没有说完,有朋友就会脱口而出:ViewPager+ Fragment实现。说起..._viewpager notifydatasetchanged

unity鼠标右键按住不放_在Windows中如何在不按住鼠标键的情况下突出显示和拖放...-程序员宅基地

文章浏览阅读775次。unity鼠标右键按住不放If you use a touchpad or trackpad, or if you have arthritis or other problems when using a mouse, you may find it difficult to hold the primary mouse button down and move the mouse at the..._untiy3d 鼠标右键一直按着

随便推点

计算机考试网上报名系统-程序员宅基地

文章浏览阅读732次,点赞27次,收藏20次。目 录(一)计算机等级考试发展状况与趋势……………………………………………………1(二)开发系统的意义………………………………………………………………………1(三)用户群及特点…………………………………………………………………………1二、系统分析………………………………………………………………………………………2(一)系统要达到的目的……………………………………………………………………2(二)系统可行性分析………………………………………………………………………2(三)业务流程分析………

linux C应用开发_linux应用开发-程序员宅基地

文章浏览阅读4.4k次,点赞2次,收藏34次。linux应用开发_linux应用开发

iPhone 4 Cydia使用教程!精选Cydia源!cydia怎么添加源!Cydia源使用方法!越狱后使用cydia全攻略!_ihpone4里cydia软件源游戏-程序员宅基地

文章浏览阅读4.3k次。转载自:http://hi.baidu.com/tyc6982/blog/item/7793eb18c9071a1635fa4191.html  2008年11月19日18:40许,iPhone中文网Cydia软件源正式上线(源地址为:iphone.tgbus.com/cydia)。这次Cydia源推出的目的主要是为了给一些WiFi用户提供方便。Cydia源中提供了一些像OpenSSH_ihpone4里cydia软件源游戏

sklearn计算余弦相似度_from sklearn.metrics.pairwise import cosine_simila-程序员宅基地

文章浏览阅读5.4k次,点赞4次,收藏16次。余弦相似度在计算文本相似度等问题中有着广泛的应用,scikit-learn中提供了方便的调用方法第一种,使用cosine_similarity,传入一个变量a时,返回数组的第i行第j列表示a[i]与a[j]的余弦相似度例:from sklearn.metrics.pairwise import cosine_similaritya=[[1,3,2],[2,2,1]]cosine_s..._from sklearn.metrics.pairwise import cosine_similarity

NV21 to NV12(YUV420SP)_nv21tonv12-程序员宅基地

文章浏览阅读1.5w次。setPreviewFormat(ImageFormat.NV21)NV21 颜色空间排列 :YYYYYYYY VUVU在用MediaCodec编码的时候,如果设置颜色空间为YUV420SP,那么则需要转换一下,YUV420SP颜色排列顺序为:YYYYYYY UVUV多说一下,YUV420 是于NV12对应的,但是5.0一下的安卓手机支持这个预览颜色的不多则需要将VU顺序进行转_nv21tonv12

Java并发编程: TransmittableThreadLocal实现父子线程之间值传递_transmittablethreadlocal父子线程数据传递-程序员宅基地

文章浏览阅读1k次,点赞23次,收藏18次。TransmittableThreadLocal 是 Alibaba 开源框架 transmittable-thread-local 中的一个核心类,它扩展了 Java 的标准 ThreadLocal 类。与标准的 ThreadLocal 不同,TransmittableThreadLocal 的值可以在线程之间传递,尤其是在线程池中的线程复用场景下。_transmittablethreadlocal父子线程数据传递

推荐文章

热门文章

相关标签