o cCs$t |¡}t |¡|||}|S)N)r.ÚabsÚtanh)r"rgÚepsÚmask_magr*r*r+rWvs zCascadedNet.bounded_maskcCsR| |¡}|jdkr'|dd…dd…dd…|j|j …f}| ¡ddks'J‚|S)Nrr )rhrFrU)r"r0rgr*r*r+Ú predict_mask{s  &zCascadedNet.predict_maskcCsh| |¡}tj||gdd|}|jdkr2|dd…dd…dd…|j|j …f}| ¡ddks2J‚|S)Nr r,rr )rhr.r/rFrU)r"r0rgÚpredr*r*r+Úpredict„s  &zCascadedNet.predict)r=r>F)ri) r8r9r:rrhrWrnrpr;r*r*r(r+r<,s # % r<) r.rÚtorch.nn.functionalÚ functionalrYÚlibrÚModulerr<r*r*r*r+Ús   $