def _loss_function(model, prediction, target): inner_loss = T.abs_(target - prediction) outer_loss = 1.0 - inner_loss combined_loss = T.stack([inner_loss, outer_loss], axis=1) return T.min(combined_loss, axis=1)