public static Pair<INDArray, INDArray> mergeLabels(INDArray[] labelsToMerge, INDArray[] labelMasksToMerge) {
        Preconditions.checkNotNull(labelsToMerge[0], "Cannot merge data: Encountered null labels array");
        int rankFeatures = labelsToMerge[0].rank();

        switch (rankFeatures) {