class_docstr: Transformer encoder 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.model.set_input_tensor(input_tensor)
  
imports: import torch
  import torch.nn.functional
  from einops import rearrange, repeat
  from nemo.collections.nlp.modules.common.megatron.module import MegatronModule
  from nemo.collections.nlp.modules.common.megatron.position_embedding import RotaryEmbedding
  from nemo.collections.nlp.modules.common.megatron.transformer import ParallelTransformer
  from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults, build_attention_mask_3d
label_desc: Use image node_id 2 for calling the MegatronRetrievalTransformerEncoderModule obj's underlying member method code with example usage: obj.set_input_tensor(input_tensor) without return types