代码实现dropout 使用nn.Dropout()初始化dropout层,由于在训练时dropout会随机丢弃而预测时不会,所以在调用model(input)前要使用model.train()或者model.eval()声明现在是在训练还是预测