code: def hybrid_forward(self, F, pred, label):  # pylint: disable=arguments-differ
          
          pred = F.split(pred, axis=2, num_outputs=2)
          start_pred = pred[0].reshape((0, -3))
          start_label = label[0]
          end_pred = pred[1].reshape((0, -3))
          end_label = label[1]
          return (self.loss(start_pred, start_label) + self.loss(
              end_pred, end_label)) / 2
docs: Parameters
          ----------
          pred : NDArray, shape (batch_size, seq_length, 2)
              BERTSquad forward output.
          label : list, length is 2, each shape is (batch_size,1)
              label[0] is the starting position of the answer,
              label[1] is the ending position of the answer.
  
          Returns
          -------
          outputs : NDArray
              Shape (batch_size,)