2025年11月26日
pytorch神经网络-一个简单例子的分解
摘要： import torch import torch.nn as nn import torch.optim as optim # 定义神经网络 class SimpleNN(nn.Module): ''' fc1 的输出维度是 128，这个张量直接作为 fc2 的输入，因此需要让 fc2 的 in_ 阅读全文
posted @ 2025-11-26 15:25 c++c鸟 阅读(6) 评论(0) 推荐(0)