Function mask2d

AutoML/darts-master/rnn/utils.py:87–92

(B, D, keep_prob, cuda=True)

def mask2d(B, D, keep_prob, cuda=True):
    m = torch.floor(torch.rand(B, D) + keep_prob) / keep_prob
    m = Variable(m, requires_grad=False)
    if cuda:
        m = m.cuda()
    return m