首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
  • 您找到你想要的搜索结果了吗?
    是的
    没有找到

    pytorch 构建神经网络

    ----神经网络由对数据执行操作的层或模块组成。torch.nn命名空间提供了构建神经网络所需的所有模块。PyTorch中的每个模块都是 nn.Module 的子类。...神经网络本身也是一个模块,但它由其他模块(层)组成。这种嵌套结构允许轻松构建和管理复杂的架构。在接下来的部分中,我们将构建一个神经网络来对 FashionMNIST 数据集中的图像进行分类。...定义模型类我们通过子类化定义我们的神经网络nn.Module,并在__init__中初始化神经网络层。每个nn.Module子类都在forward方法中实现对输入数据的操作。...它们在线性变换后以引入非线性变换,帮助神经网络学习各种现象。在这个模型中,我们在线性层之间使用nn.ReLU,但是还有其他激活可以在模型中引入非线性。...模型参数神经网络内的许多层都是参数化的,即在训练期间优化的相关权重和偏差。

    39830
    领券