千千文学网

第623章 这框架简直是神器（第2页）

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

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

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

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

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

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

直到看到文档里清清楚楚地写著：该模块通过卷积核提取局部特徵，极为適用於图像识別与计算机视觉领域。

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

“我们下午吃饭的时候，討论的李飞飞教授那个imagenet数据集，不就是图像分类吗？”

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

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

两人对著这些功能模块，越討论越兴奋。

“一航，別光看著眼馋了，纸上得来终觉浅。”郭长征提议。

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

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

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

在以往，用英伟达的cuda写代码，繁琐。

程式设计师必须手动在主机的內存里开闢空间，再在显卡的显存里开闢空间。

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

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

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

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

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

不到二十行代码。

简洁，优雅。

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

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

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