def METHOD_NAME(self, batch_data, text_len, neg_size):
    text = paddle.to_tensor(batch_data[0].numpy().astype('int64').reshape(
        -1, text_len))
    pos_tag = paddle.to_tensor(batch_data[1].numpy().astype('int64')
                               .reshape(-1, 1))
    neg_tag = paddle.to_tensor(batch_data[2].numpy().astype('int64')
                               .reshape(-1, neg_size))
    return [text, pos_tag, neg_tag]