前些日子太忙,停止打卡了好久,努力补上进度。
今天学习的是第六课网络构建,涉及到神经网络模型,即对神经网络结构进行构建和管理。
首先是定义模型类,第一神经网络时,可以继承nn.Cell类,在_init_进行子cell的实例化和状态管理,在construct方法中实现Tensor操作。
实例化nn.Flatten层,28×28的2D张量转化为784大小的连续数组。
nn.Dense为全连接层,其使用权重和输入进行线性变换。
nn.ReLU层网络中加入非线性的激活函数,帮助神经网多学习各种复杂特征。