def sample_mask(self): keep = 1.0 - self.dropout self.mask = V(th.bernoulli(T(1, self.hidden_size).fill_(keep)))