FCN Implementation : Loss Function

[ptrblck]

nn.NLLLoss now takes multi-dim input. Have a look at the docs for the shape information.