Pytorch Medical Image Segmentation Using Squeeze And Expansion