神经网络由对数据执行操作的层/模块组成。torch.nn 提供了构建神经网络所需的所有模块。 ... ReLU() ) def forward(self, x): #前向传播 x = self.flatten(x) logits ...
確定! 回上一頁