code: def __init__(self, in_subs, out_sub, out_shape):
      self.in_subs = in_subs
      self.out_sub = out_sub
      self.out_shape = out_shape
  
imports: import warnings
  import chainer
  from chainer import backend
  from chainer.backends import cuda
  from chainer import function_node
  from chainer import utils
  from chainer.utils import argument
  from chainer.utils import type_check
label_desc: Use image node_id 1 to create a new DiagEinSum object from inherited base classes: EinSum with example: obj = DiagEinSum(in_subs, out_sub, out_shape)