code: def get(self, lengths: Union[mx.nd.NDArray, int, float]) -> Union[mx.nd.NDArray, float]:
          
          return self.hybrid_forward(None, lengths)
docs: Calculate the length penalty for the given vector of lengths.
  
          :param lengths: A scalar or a matrix of sentence lengths of dimensionality (batch_size, 1).
          :return: The length penalty. A scalar or a matrix (batch_size, 1) depending on the input.