Dropout — PyTorch 1.10.1 documentation
pytorch.org › generated › torchDropout. During training, randomly zeroes some of the elements of the input tensor with probability p using samples from a Bernoulli distribution. Each channel will be zeroed out independently on every forward call. This has proven to be an effective technique for regularization and preventing the co-adaptation of neurons as described in the ...
torch.masked_select — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.masked_select. Returns a new 1-D tensor which indexes the input tensor according to the boolean mask mask which is a BoolTensor. The shapes of the mask tensor and the input tensor don’t need to match, but they must be broadcastable. input ( Tensor) – the input tensor. out ( Tensor, optional) – the output tensor.