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)

 

posted @ 2018-11-13 16:53  勿忘初心233  阅读(2500)  评论(0)  编辑  收藏  举报