MSE           = ((pred - target) ** 2).mean()
CrossEntropy  = -log( softmax(logits)[correct_class] )