Implement Focal Loss | TensorTonic