class_docstr: Vision Transformer Model.
class_docstr_tok: ["Vision","Transformer","Model","."]
docstr: See megatron.model.transformer.set_input_tensor()
docstr_tok: ["See","megatron.model.transformer.set_input_tensor","(",")"]
code: def set_input_tensor(self, input_tensor):
      """See megatron.model.transformer.set_input_tensor()"""
      self.transformer.set_input_tensor(input_tensor)
  
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 2 for calling the VitBackbone obj's underlying member method code with example usage: obj.set_input_tensor(input_tensor) without return types