售前咨询:400-0666-318
PyTorch的神经网络模块是torch.nn,它提供了用于构建神经网络的模块和函数。这个模块包含了各种神经网络层(如全连接层,卷积层,循环神经网络层等),以及损失函数和优化器等工具,可以帮助用户构建和训练神经网络模型。PyTorch的神经网络模块使得构建和训练神经网络变得更加简单和方便。
torch.nn
辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读: PyTorch中如何实现自监督学习