lddmm_pytorch.py 文件源码

def _vertices_to_measure( q, connec ) :
        """
        Transforms a torch array 'q1' into a measure, assuming a connectivity matrix connec.
        It is the Torch equivalent of 'to_measure'.
        """
        a = q[connec[:,0]] ; b = q[connec[:,1]]
        # A curve is represented as a sum of diracs, one for each segment
        x  = .5 * (a + b)                     # Mean
        mu = torch.sqrt( ((b-a)**2).sum(1) )  # Length
        return (x, mu)