Solved by SimonW in post #2 
              
              
                The dim kwarg is implemented after 0.2.0 release. Therefore, if you didn’t install from source, you likely don’t have that available right now. Please refer to this doc on 0.2.0 http://pytorch.org/docs/0.2.0/_modules/torch/nn/modules/activation.html#LogSoftmax. 
The default dimension selection is so…