code: def __init__(self, operands, argmax):
      self.operands = operands
      self.argmax = argmax
  
file_path: pyro/pyro/ops/einsum/torch_map.py
imports: import operator
  from functools import reduce
  from pyro.ops import packed
  from pyro.ops.einsum.adjoint import Backward, einsum_backward_sample, transpose, unflatten
  from pyro.ops.einsum.util import Tensordot
label_desc: Use image node_id 1 to create a new _EinsumBackward object from inherited base classes: Backward with example: obj = _EinsumBackward(operands, argmax)