code: def kaiming_normal_(
      tensor, a=0, mode="fan_in", nonlinearity="leaky_relu"
  ):
      """Drop-in replacement of `torch.nn.init.kaiming_normal_`.
      Note:
          -  if using this function, ensure `a` does not depend on fan-in,
             fan-out, or other notions of width, e.g. if a = 0.
          - `tensor` should have `infshape` attribute set by `set_base_shapes`.
      """
      if 0 in tensor.shape:
          warnings.warn("Initializing zero-element tensors is a no-op")
          return tensor
      fan = _calculate_correct_fan(tensor, mode)
      gain = calculate_gain(nonlinearity, a)
      std = _inf_fan_adjust_kaiming(gain / math.sqrt(fan), tensor, mode)
      with torch.no_grad():
          return tensor.normal_(0, std)
  
file_path: NeMo/nemo/collections/nlp/modules/common/megatron/mup/init.py
imports: import math
  import warnings
  import torch
  from torch.nn.init import _calculate_correct_fan, _calculate_fan_in_and_fan_out, _no_grad_fill_, _no_grad_normal_, _no_grad_uniform_, calculate_gain
label_desc: Use image node_id 10 for calling a global function with example usage: kaiming_normal_(tensor, a, mode, nonlinearity) and returns: tensor, tensor