class_docstr: Context Gating Layer
  Args:
      dimension (int): dimension
      add_batch_norm (bool): add batch normalization
code: def __init__(self, dimension, add_batch_norm=True):
      super().__init__()
      self.fc = nn.Linear(dimension, dimension)
      self.add_batch_norm = add_batch_norm
      self.batch_norm = nn.BatchNorm1d(dimension)
  
imports: import os
  import torch
  import torch.nn.functional
  from typing import Any, List, Dict
  from collections import OrderedDict
  from torch import nn
  from towhee.models.mdmmt.bert_mmt import BertMMT
label_desc: Use image node_id 1 to create a new ContextGating object from inherited base classes: nn with example: obj = ContextGating(dimension, add_batch_norm)