PyTorch的神经网络模块是torch.nn,它提供了用于构建神经网络的模块和函数。这个模块包含了各种神经网络层(如全连接层,卷积层,循环神经网络层等),以及损失函数和优化器等工具,可以帮助用户构建和训练神经网络模型。PyTorch的神经网络模块使得构建和训练神经网络变得更加简单和方便。
torch.nn