code: def initialize_tensors(self):
      # Set the tensors from policy (user module) to container (DS module)
      self.set_attention(*self.policy.attention())
      self.set_mlp(self.config.moe.type)
      self.set_layernorm(*self.policy.layernorm())
  
file_path: DeepSpeed/deepspeed/module_inject/containers/base_moe.py
imports: from .base import *
  from deepspeed import comm
  import deepspeed.ops.transformer
  from deepspeed.accelerator import get_accelerator
label_desc: Use image node_id 3 for calling the BaseTransformerMoEContainer obj's underlying member method code with example usage: obj.initialize_tensors() without return types