神经网络中_,predicted=中_,的作用

在神经网络训练中，经常会用到这样一段代码。

_,predicted=torch.max(output.data,dim=1)

其中_,是什么作用呢？
output是通过神经网络最后一层softmax函数作用之后的输出。例如，假设我们在分类问题中共有4类，分别是0，1，2，3。
假设output.data的第一行是[1,0,0,0]。（表明属于第一个类别0的概率是1)。
那么，假设不加_,代码

predicted=torch.max(output.data,dim=1)

输出的则是 1。（其中1是tensor)。加了 , 之后，输出的则变成了0。
因此，, 的作用是，是使predicted返回output.data行中最大数值所在位置代表的类别。如本例中代表所属类别0。

神经网络中_,predicted=中_,的作用相关推荐

神经网络中神经元是什么_是什么使神经网络脆弱
神经网络中神经元是什么 What do the images below have in common? 以下图片有什么共同点? Most readers will quickly catch on ...
文件单片机_单片机C语言编程中reg52.h头文件的作用
前言:本人出于爱好将不定期发送电气电工.前端.单片机等内容,可能会无法顾及关注我的所有人需求,请大家按需收藏自己想要知识,有用则收之,无用则弃之,不系统更新,仅供零星学习O(∩_∩)O哈哈~ 在代码的 ...
matlab 神经网络设计多层隐含层_[rbf神经网络]请教关于MATLAB中RBF神经网络隐含层神经元个数的问题...
rbf神经网络请教关于MATLAB中RBF神经网络隐含层神经元个数的问题问题补充:MATLAB中用newrb(P, T, goal, spread)创建的神经网络,用于数据的预测,程序可以运行并能 ...
cmenu 隐藏子项中的一个子项_区分Maven中dependencyManagement与dependencies的作用
导读:使用maven是为了更好的帮项目管理包依赖,maven的核心就是pom.xml.而maven中有许多的标签,下面我们主要讨论parent.dependencies与dependencyManag ...
比较两组数据的差异用什么图更直观_芯片数据分析中常见的一些图的作用
今天给大家讲讲芯片数据分析中常见的一些图的作用,让大家伙儿知道它们在BB些啥. 箱式图(Box plot) 基因芯片的原始数据是需要进行标准化处理的,主要目的是消除由于实验技术(如荧光标记效率.扫描参 ...
什么是微信防火墙_【听课】第4节什么是“合并单元”(MU)？智能站中的合并单元有什么作用？...
完整视频,点击文末一.会员权益 "视频+PPT"讲解,视频播放+赠送课件 1.会员免费任意听 2.会员赠送课件(进入群聊,@群主并加小编微信获取) 3.支持回听,可全屏,可回看, ...
Java main方法_解释Java中的main方法，及其作用_一个java文件中可包含多个main方法
public static void main(String[] args) {} 或者 public static void main(String args[]) {} main方法是我们学习Ja ...
管理沟通中移情的应用_移情在设计中的重要性
管理沟通中移情的应用 One of the most important aspects of any great design is the empathetic understanding of ...
大屏设计的视觉统一_视觉设计中的统一
大屏设计的视觉统一视觉设计的统一性是什么? (What is unity in visual design?) The concept of unity in visual design means ...
mardown 标题带数字_标题中带有数字的故事更成功吗？
mardown 标题带数字统计 (Statistics) I have read a few stories on Medium about writing advice, and there we ...

神经网络中_,predicted=中_,的作用

神经网络中_,predicted=中_,的作用

神经网络中_,predicted=中_,的作用相关推荐

最新文章

热门文章