@ptrblck,
The documentation for 0.3.1 says the shape needs to be the same -> It shouldn’t work but it does.
I would also suggest that when the last layer output is single output PyTorch output will be a vector and not a single column matrix.
PyTorch 0.4.0 is released on PyTorch channel not on Anaconda’s official channel.
I always ask them: https://github.com/ContinuumIO/anaconda-issues/issues/7410 4
Hopefully they will update it soon.