diffalign.models.epsnet.diffusion.get_distance

diffalign.models.epsnet.diffusion.get_distance(pos: Tensor, edge_index: Tensor) → Tensor