def hybrid_forward(self, F, pred, label):  
pred = F.split(pred, axis=<NUM_LIT:2>, num_outputs=<NUM_LIT:2>)<EOL>start_pred = pred[<NUM_LIT:0>].reshape((<NUM_LIT:0>, -<NUM_LIT:3>))<EOL>start_label = label[<NUM_LIT:0>]<EOL>end_pred = pred[<NUM_LIT:1>].reshape((<NUM_LIT:0>, -<NUM_LIT:3>))<EOL>end_label = label[<NUM_LIT:1>]<EOL>return (self.loss(start_pred, start_label) + self.loss(<EOL>end_pred, end_label)) / <NUM_LIT:2><EOL>
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,)