class_docstr: Planar Flow model.
  
  Args:
      model_config (PlanarFlowConfig): The PlanarFlow model configuration setting the main parameters of
          the model.
code: def __init__(self, model_config: PlanarFlowConfig):
      BaseNF.__init__(self, model_config)
  
      self.w = nn.Parameter(torch.randn(1, self.input_dim))
      self.u = nn.Parameter(torch.randn(1, self.input_dim))
      self.b = nn.Parameter(torch.randn(1))
      self.activation = ACTIVATION[model_config.activation]
      self.activation_derivative = ACTIVATION_DERIVATIVES[
          model_config.activation
      ]
      self.model_name = "PlanarFlow"
  
      nn.init.normal_(self.w)
      nn.init.normal_(self.u)
      nn.init.normal_(self.b)
  
file_path: benchmark_VAE/src/pythae/models/normalizing_flows/planar_flow/planar_flow_model.py
imports: import os
  import torch
  import torch.nn
  import torch.nn.functional
  from ...base.base_utils import ModelOutput
  from ..base import BaseNF
  from .planar_flow_config import PlanarFlowConfig
label_desc: Use image node_id 1 to create a new PlanarFlow object from inherited base classes: BaseNF with example: obj = PlanarFlow(model_config)