多层感知机的简洁实现

现在还是在用cpu训练,要使用gpu的话需要把模型等to devices(cuda)