code: def run(self, batch: mx.io.DataBatch) -> List[mx.nd.NDArray]:
          
          self.module.forward(batch, is_train=False)
          return self.module.get_outputs()
docs: Runs the forward pass and returns the outputs.
  
          :param batch: The batch to run.
          :return: The grouped symbol (probs and target dists) and lists containing the data names and label names.