code: def __init__(self):
      super().__init__()
      self._ctx = None
      self._fw_suffix = "agglayer_fw_{}"
      self._bw_suffix = "agglayer_bw_{}"
      self._pred_suffix = "agglayer_pred_{}"
      self._fw_count = 0
      self._bw_count = 0
      self._pred_count = 0
      self._has_ctx = False
      self._model = None
      self.training = True
      self.device = None
  
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 _AggLayerBase object from inherited base classes: t.nn with example: obj = _AggLayerBase()