point_edge_dist_matrix#

iskra.geometry.distances.point_edge_dist_matrix(x: Tensor, edges: Tensor) Tensor[SOURCE]#

Pairwise distances between a set of points and a set of edges.

Parameters:
  • x (Tensor[Float, [Bs1, N, Dim]]) – Set of points.

  • edges (Tensor[Float, [Bs2, M, 2, Dim]]) – Set of edges.

Returns:

(Tensor[Float, [Bs3, N, M]]) – Pairwise point-edge distances.