code: def get_loss(self, pred: mx.sym.Symbol, labels: mx.sym.Symbol) -> mx.sym.Symbol:
          
          labels = mx.sym.reshape(labels, shape=(-1, 1))
          loss_value = self.loss_config.length_task_weight / 2 * mx.sym.square(pred - labels)
          loss_value = mx.sym.MakeLoss(data=loss_value,
                                       normalization='batch',
                                       name=self.name)
          return loss_value
docs: Returns MSE loss and output symbol given logits and expected integers as labels.
  
          :param pred: Predictions. Shape: (batch_size, 1).
          :param labels: Targets. Shape: (batch_size,).
          :return: Loss symbol.