def forward(self, X): X = super(Dropout, self).forward(X) eps = torch.Tensor(*X.size()) eps.fill_(self.p) eps = Variable(torch.bernoulli(eps)) return X * eps