重生08,豆包成了我的外挂

第623章 这框架简直是神器!(2/3)

天才一秒记住【倩玉小说网】地址:https://m.qianyuwj.com

首发:~第623章 这框架简直是神器!

“郭老师,你看这个,这里封装了好多高级模块。”

楚一航点开了一个名为卷积的目录。

里面列着各种奇怪的函数名。

两人仔细看着文档的解释。

里面有一个叫卷积神经网络的封装层。

两人平时研究的都是些传统的机器学习算法,对这种结构并不算特别清楚。

直到看到文档里清清楚楚地写着:该模块通过卷积核提取局部特征,极为适用于图像识别与计算机视觉领域。

“图像识别?”郭长征敏锐地抓住了这个词。

“我们下午吃饭的时候,讨论的李飞飞教授那个ImageNet数据集,不就是图像分类吗?”

楚一航一拍桌子:“绝了!连这种专门针对图像的模块都提前封装好了。”

“这简直就是给咱们量身定制的武器库啊。”

两人对着这些功能模块,越讨论越兴奋。

“一航,别光看着眼馋了,纸上得来终觉浅。”郭长征提议。

“咱们现在就动手,按这个文档,写一个简单的几层网络试试水。”

“好嘞。”楚一航立刻切换到代码编辑器。

按照平时用CUDA编写底层算力代码的习惯,楚一航习惯性地准备先写一堆内存分配的语句。

在以往,用英伟达的CUDA写代码,繁琐。

程序员必须手动在主机的内存里开辟空间,再在显卡的显存里开辟空间。

然后要把数据从主机复制到显卡,接着手动设置线程块和网格的大小。

稍微设置不合理,显卡就直接罢工不干。

算完之后,还得再把数据从显卡慢吞吞地拷回主机。

但当楚一航按照TensorFlow的文档指引敲击键盘时,他发现那些全都不需要了。

他只需要定义好数据格式,声明好网络层数,框架的底层执行引擎会自动接管所有的硬件资源调度。

不到二十行代码。

简洁,优雅。

“这就写完了?”楚一航自己都不敢相信。

“这几十行代码,放以前用C语言加CUDA手搓,至少得写大半天,还得调试一整天。”

郭长征凑在屏幕前,反反复复检查了三遍代码逻辑。

没钱又任性提示您:看后求收藏(倩玉小说网https://m.qianyuwj.com),接着再看更方便。

人气小说推荐More+

混沌天帝诀凌峰苏琳
混沌天帝诀凌峰苏琳
身怀至尊天子之血,开启最强天子之眼!  武技复制!嗜血狂暴!看破虚幻!记忆读取!焚世黑炎!分身瞬移!空间粉碎!无限视界!时间静止!……  【混沌天帝】凌峰:“我凭这双眼,敢叫天地颤栗!”
剑轻阳
人在斗破:天赋绝世竟带系统
人在斗破:天赋绝世竟带系统
穿越斗破苍穹,觉醒前世记忆。没有随身老爷爷。该如何在这大陆上崛起?机缘巧合,觉醒抽取系统。【滴…...
仪昔
修仙界只有妖女了是吗
修仙界只有妖女了是吗
“你觉得修仙什么最重要?”多年以后,当已经证道成仙的路长远翻开剑谱,准备书写自己的成仙感悟之时,遇见...
月下千早
祭祀百年,我成了部落先祖
祭祀百年,我成了部落先祖
穿越到大荒,沈灿因为身体‘孱弱‘成为炙炎部落祖庙的守祧(tiao),负责祖庙日常洒扫,香火供奉,日常祭祀...
山人有妙计
朕,都是为了大汉!
朕,都是为了大汉!
后世的历史学家虽然都认同刘邈三兴汉室的功绩,但同时也对刘邈的某些陋习予以批评。对此,刘邈当然有话要...
一天写三章