baby.utils.split_batch_pred

baby.utils.split_batch_pred(outputs)