档里推荐了一个叫ReLU的函数。”“ReLU?我找找……有了,一行代码搞定。”在搭建的过程中,他们也不是一帆风顺。时不时会遇到一些报错。比如“张量维度不匹配”这种经典的错误。但得益于框架清晰的报错提示,他们总能迅速定位到是哪一层的矩阵乘法出了问题。两个人为了算清楚上一层的输出怎么和下一层的输入对齐,甚至找了张A4纸在桌子上画起了矩阵图。不知不觉中,时间过得飞快。楚一航敲击回车,最后一行代码编写完毕。他们把整个神经网络的模型类封装好了,并且配置好了损失函数和优化器。楚一航伸了个大大的懒腰,只觉得脖子和后背一阵酸痛。郭长征也揉了揉酸涩的眼睛。窗外传来了鸟叫声。楚一航听到后愣了一下,瞥了一眼屏幕右下角的时间,整个人愣住了。“郭老师……五点半了。”