code: def __init__(
      self, merge_type: Literal["sum", "concat"] = "sum", concat_dim=1
  ):
      super(AggLayerGuest, self).__init__()
      self._host_input_caches = None
      self._merge_type = merge_type
      assert (
          self._merge_type in MERGE_TYPE
      ), f"merge type should be one of {MERGE_TYPE}"
      assert isinstance(concat_dim, int), "concat dim should be int"
      self.register_buffer("_concat_dim", torch.LongTensor(concat_dim))
  
imports: import torch
  from typing import List, Literal
  import torch
  from fate.arch import Context
  from torch.nn.modules.module import T
label_desc: Use image node_id 1 to create a new AggLayerGuest object from inherited base classes: _AggLayerBase with example: obj = AggLayerGuest(merge_type, concat_dim)