Tensorflow Custom Loss Function Example