Mlp Pytorch Example