what the shapes are in pred and label of distribution_focal_loss