logit和labels最后一个维度匹配不上，求大神指教！
这几天在运行一个论文中别人改进的3D U-net，原代码是解决多尺度输入的4分类问题。我想把它改成单尺度的二分类也就是分割，但是在运行的时候，卡在了logist和label最后一个维度不匹配的错误上，希望大神可以救救孩子吧！(错误如下图)