Hannes Kuchelmeister
Add code so loss function uses torch.Size([x,1]) instead of torch.Size([x])
ba9c868