class_docstr: Constructes a diagonal Gaussian distribution from the hidden states and samples from it using reparametrization.
class_docstr_tok: ["Constructes","a","diagonal","Gaussian","distribution","from","the","hidden","states","and","samples","from","it","using","reparametrization","."]
docstr: Provide here all required inputs
docstr_tok: ["Provide","here","all","required","inputs"]
code: def input_names(self):
      """
      Provide here all required inputs
      """
      return ["hiddens", "hiddens_mask"]
  
file_path: NeMo/nemo/collections/nlp/modules/common/megatron/hiddens/megatron_hidden_transform.py
imports: import math
  import torch
  from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults, init_method_normal
label_desc: Use image node_id 2 for calling the MegatronGaussianHiddenTransform obj's underlying member method code with example usage: obj.input_names() and returns: list