Add shape checking to Accuracy and Loss functions

[kmheckel]

It's a common mistake to get the output shape wrong compared to the target label shape, adding some kind of check in the loss and acc functions to throw an error would be useful.