class_docstr: https://arxiv.org/abs/2212.00794
class_docstr_tok: ["https",":","\/\/arxiv.org\/abs\/2212.00794"]
code: def __init__(
      self, prob, class_token_length=8, exclude_cls_tokens=True
  ):
      assert 0 <= prob < 1.0
      super(DropPatch, self).__init__()
      self.prob = prob
      self.class_token_length = class_token_length
      self.exclude_cls_tokens = exclude_cls_tokens
  
imports: import math
  from functools import partial
  import einops
  import torch
  import torch.nn.functional
  from nemo.collections.nlp.modules.common.megatron.fused_layer_norm import get_layer_norm
  from nemo.collections.nlp.modules.common.megatron.module import MegatronModule
  from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults, init_method_normal, scaled_init_method_normal
  from nemo.collections.vision.modules.common.megatron.vision_transformer import ParallelVisionTransformer
label_desc: Use image node_id 1 to create a new DropPatch object from inherited base classes: MegatronModule with example: obj = DropPatch(prob, class_token_length, exclude_cls_tokens)