def logit2prob()

in submix.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

def logit2prob(logit):
    return torch.exp(logit)/(torch.sum(torch.exp(logit)))