2024年6月25日
ChatGPT学习笔记之Linear层和ReLU 激活函数
摘要： 有一个简单的学习模型 class SimpleModel(nn.Module): def __init__(self): super(SimpleModel, self).__init__() self.fc1 = nn.Linear(4, 16) # 输入4维，隐藏层16维 self.fc2 = 阅读全文
posted @ 2024-06-25 16:27 阿坦 阅读(59) 评论(2) 推荐(1) 编辑