class_docstr: Output layer for doc regression models. Currently only supports Mean Squared Error
  loss.
  
  Args:
      loss (MSELoss): config for MSE loss
      squash_to_unit_range (bool): whether to clamp the output to the range [0, 1],
          via a sigmoid.
code: def __init__(
      self, loss_fn: MSELoss, squash_to_unit_range: bool = False
  ) -> None:
      super().__init__()
      self.loss_fn = loss_fn
      self.squash_to_unit_range = squash_to_unit_range
      log_class_usage(__class__)
  
imports: from typing import Any, Dict, List, Optional, Tuple, Union
  import torch
  from pytext.config.component import create_loss
  from pytext.loss import MAELoss, MSELoss
  from pytext.utils.usage import log_class_usage
  from .output_layer_base import OutputLayerBase
label_desc: Use image node_id 2 to create a new RegressionOutputLayer object from inherited base classes: OutputLayerBase with example: obj = RegressionOutputLayer(loss_fn, squash_to_unit_range)