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 generated outputs
docstr_tok: ["Provide","here","all","generated","outputs"]
code: def output_names(self):
      """
      Provide here all generated outputs
      """
      return ["z_mean", "z_logvar", "z", "z_log_prob"]
  
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 3 for calling the MegatronGaussianHiddenTransform obj's underlying member method code with example usage: obj.output_names() and returns: list