class_docstr: Based on <https://arxiv.org/abs/2003.02645>
  Implements A-MIM loss with a unit Normal anchor.
  A-MIM - asymmetric MIM (without sampling)
class_docstr_tok: ["Based","on","<","https",":","\/\/arxiv.org\/abs\/2003.02645",">","Implements","A-MIM","loss","with","a","unit","Normal","anchor",".","A-MIM","-","asymmetric","MIM","(","without","sampling",")"]
docstr: Add here all required inputs
docstr_tok: ["Add","here","all","required","inputs"]
code: def _input_names(self):
      """Add here all required inputs"""
      return ["z", "z_log_prob"]
  
imports: import math
  import torch
label_desc: Use image node_id 2 for calling the MegatronAMIMHiddenLoss obj's underlying member method code with example usage: obj._input_names() and returns: list