def METHOD_NAME(self, batch_data, config):
    sparse_tensor = []
    for b in batch_data:
        sparse_tensor.append(
            paddle.to_tensor(b.numpy().astype('int64').reshape(-1, 1)))
    label = sparse_tensor[-1]
    return label, sparse_tensor[:-1]