Garfield.modules.compute_contrastive_clusterloss

Garfield.modules.compute_contrastive_clusterloss(c_i, c_j, class_num, temperature)

Cluster loss function.

Args:

c_i (torch.Tensor): First set of cluster probabilities. c_j (torch.Tensor): Second set of cluster probabilities. class_num (int): Number of classes. temperature (float): Temperature scaling factor. device (torch.device): The device to perform computations on.

Returns:

torch.Tensor: The computed loss value.