tensorplay.nn.modules

tensorplay.nn 模块提供了构建神经网络的基础组件。它的设计高度模块化，并与 PyTorch 的 nn API 兼容。

核心概念

• Module：所有神经网络层的基类。你的模型也应该继承此类。
• Parameter：一种特殊的 Tensor，当作为属性赋值时，会自动注册为模块参数。
• Functional：层的无状态版本，位于 tensorplay.nn.functional 中。

模块分类

具体层请参阅：

• 模块 (基类)：基类如 Module 和 Parameter。
• 线性层：全连接层。
• 卷积层：1D、2D 和 3D 卷积层。
• 池化层：最大池化和平均池化。
• 激活函数：ReLU, Sigmoid, Tanh 等。
• 归一化层：BatchNorm, LayerNorm, InstanceNorm。
• 损失函数：MSE, CrossEntropy, NLLLoss。
• Dropout 层：Dropout 和 AlphaDropout。
• 容器：Sequential, ModuleList, ModuleDict。