def METHOD_NAME(self, z_batch: "tf.Tensor", lambda_hy: float, z_trigger: np.ndarray, x_target: np.ndarray):
    """
    The loss function used to perform a trail attack
    :param z_batch: triggers to be trained on
    :param lambda_hy: the lambda parameter balancing how much we want the auxiliary loss to be applied
    """
    import tensorflow as tf
    return lambda_hy * tf.math.reduce_mean(
        tf.math.squared_difference(
            tf.dtypes.cast(self.estimator.model(z_trigger), tf.float64),
            tf.dtypes.cast(x_target, tf.float64),
        )
    ) + tf.math.reduce_mean(
        tf.math.squared_difference(
            tf.dtypes.cast(self.estimator.model(z_batch), tf.float64),
            tf.dtypes.cast(self._model_clone(z_batch), tf.float64),
        )
    )