code: def __init__(
      self, _id, branches, merge=None, in_bottle=None, out_bottle=None
  ):
      self._id = _id
      self.in_bottle = in_bottle
      self.branches = branches
      self.out_bottle = out_bottle
      self.merge = merge
      if self.merge == "add":
          out_dim = []
          for branch in self.branches:
              out_dim.append(branch.out_features_dim)
          assert np.std(out_dim) == 0, (
              "<%s> require the output dim of all branches are the same"
              % self.merge
          )
      elif self.merge is None:
          assert len(self.branches) == 1, "Invalid"
  
      self.output_op = None
  
file_path: auptimizer/src/aup/Proposer/eas/models/layer_multi_branch.py
imports: import tensorflow
  import numpy
  from .layer_cascade import LayerCascade
label_desc: Use image node_id 1 to create a new LayerMultiBranch object  with example: obj = LayerMultiBranch(_id, branches, merge, in_bottle, out_bottle)