def METHOD_NAME(self, pos_z: Tensor, neg_z: Tensor, summary: Tensor) -> Tensor:
    r"""Computes the mutual information maximization objective."""
    pos_loss = -torch.log(
        self.discriminate(pos_z, summary, sigmoid=True) + EPS).mean()
    neg_loss = -torch.log(1 -
                          self.discriminate(neg_z, summary, sigmoid=True) +
                          EPS).mean()
    return pos_loss + neg_loss