(output_fake)
return loss
r"""
    Computes the wasserstein loss for generator.

    Args:
        output_fake (Tensor): Discriminator output logits for fake images.

    Returns:
        Tensor: A scalar tensor loss output.
r"""
    Computes the wasserstein loss for generator.
r Computes the wasserstein loss for generator .
def wasserstein_loss_gen(output_fake):
    r"""
    Computes the wasserstein loss for generator.

    Args:
        output_fake (Tensor): Discriminator output logits for fake images.

    Returns:
        Tensor: A scalar tensor loss output.
    """
    loss = -output_fake.mean()

    return loss
def wasserstein_loss_gen ( output_fake ) : loss = - output_fake . mean ( ) return loss