def METHOD_NAME(self, pred):
    neg_w = torch.cat([self.map_func(self.pos_neg), pred.new_ones(1)])
    pos_w = 1 + self.alpha * (1 - neg_w)
    neg_w = neg_w.view(1, -1).expand(self.n_i, self.n_c)
    pos_w = pos_w.view(1, -1).expand(self.n_i, self.n_c)
    return pos_w, neg_w