Dropout in PyTorch – An Example