Generalized Probabilistic U-Net for medical image segementation