Jing Jiang

I am a beginner and I would like to know how TransMorph-diff obtains an inverse deformation field? Can I use this method below? ``` flow_mean = self.flow_mean(x) log_sigma = self.flow_sigma(x)...