code: def scaled_init_method_normal(sigma, num_layers):
      """Init method based on N(0, sigma/sqrt(2*num_layers)."""
      std = sigma / math.sqrt(2.0 * num_layers)
  
      def init_(tensor):
          return torch.nn.init.normal_(tensor, mean=0.0, std=std)
  
      return init_
  
imports: import math
  import operator
  from functools import reduce
  import torch
  from megatron.core import parallel_state
  from megatron.core.dist_checkpointing.mapping import ShardedTensor
label_desc: Use image node_id 11 for calling a global function with example usage: scaled_init_method_normal(sigma, num_layers) and returns: init_, torch