PyTorch学习笔记之神经网络包 nn 和优化器 optim

torch.nn 是专门为神经网络设计的模块化接口。构建于 Autograd 之上,可用来定义和运行神经网络。下面介绍几个常用的类: 注: torch.nn 为了方便使用,将它设置成 nn 的别名。 除了 nn 别名以外,我们还引用了 nn.functional,这个包中包含了神经网络中使用的一些常用的函数,这些函数的特点是,不具有可学习的参数(ReLu, pool, DropOut 等),这些函
相关文章
相关标签/搜索