def METHOD_NAME(self):
    try:
        logit = self.alphas(self.rho.weight) # torch.mm(self.rho, self.alphas)
    except:
        logit = self.alphas(self.rho)
    beta = F.softmax(logit, dim=0).transpose(1, 0) ## softmax over vocab dimension
    return beta