code: def __init__(self, input_dim: int):
      super().__init__()
      self.mlp = PositionWiseFeedForward(
          d_model=input_dim, d_ff=2 * input_dim
      )
  
file_path: pytorch_tabular/src/pytorch_tabular/ssl_models/common/ssl_utils.py
imports: import pytorch_lightning
  from torch import Tensor
  from pytorch_tabular.models.common import PositionWiseFeedForward
label_desc: Use image node_id 1 to create a new Contrastive object from inherited base classes: pl with example: obj = Contrastive(input_dim)