o  ia@sXddlmZddlmZmZddlZddlmmZ ddl m Z dgZ ddd dZ dS)) annotations)CallableSizedN)InterpolateModedefault_upsamplerF spatial_sizer align_cornersboolreturn&Callable[[torch.Tensor], torch.Tensor]csfdd}|S)z A linear interpolation method for upsampling the feature map. The output of this function is a callable `func`, such that `func(x)` returns an upsampled tensor. cs8tjtjtjg}|td}tj|t|jdS)N)sizemoder) rLINEARBILINEAR TRILINEARlenF interpolatestrvalue)x linear_mode interp_moderr\/home/dell461/cl/sdc2/last_ska_mid/HISourceFinder-master-l/src/monai/visualize/visualizer.pyupszdefault_upsampler..upr)rrrrrrrs)F)rrrr r r ) __future__rcollections.abcrrtorchtorch.nn.functionalnn functionalr monai.utilsr__all__rrrrrs