网络层

当前实现了以下网络层:

  • 卷积层
    • 有权重参数
      • 卷积层(Conv2d)
      • 全连接层(FC)
    • 无权重参数
      • 批量归一化层(BN)
      • 随机失活层(Dropout)
      • 全局平均池化层(GAP)
      • 最大池化层(MaxPool)
  • 激活函数
    • 整流线性函数(ReLU)
  • 损失函数
    • 交叉熵损失(CrossEntropyLoss)
  • 评分函数
    • Softmax