code: def __init__(self, x_size, h_sizes, z_size, non_linear_layer):
      super().__init__()
      self.z_size = z_size
      output_size = 2 * z_size
      self.mlp = MLP(x_size, h_sizes + [output_size], non_linear_layer)
  
imports: import torch
  import torch.nn
  from torch.nn.functional import softplus
label_desc: Use image node_id 1 to create a new Encoder object from inherited base classes: nn with example: obj = Encoder(x_size, h_sizes, z_size, non_linear_layer)