【深度学习基础】手动实现多层感知器MLP神经网络(向前向后传播,relu激活函数,mse损失函数)
手动实现了一个有10个隐藏层的全连接神经网络的向前与向后传播(主要是完成学校作业…TAT)import torchclass MLP_Scratch: def __init__(self, in_dim, hidden_dim, out_dim, num_hlayer): """ Args: in_dim: input data dimension hidden_dim: MLP hidden layer dim
原创
2022-02-12 20:15:26 ·
1314 阅读 ·
0 评论