serWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.

torch版本的原因，torch0.4.1使用softmax时需要加上维度参数：

p=torch.nn.functional.softmax(x,dim=1)