closest_edge

iskra.geometry.distances.closest_edge(x: Tensor, edges: Tensor) → tuple[Tensor, Tensor, Tensor][SOURCE]

Finds the closest edges to a set of points.

Bs3 is the shape that results from broadcasting Bs1 and Bs2.

Parameters:

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

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

Returns:

• ​ (Tensor[Float, [Bs3, Dim]]) – Closest projection of x onto the set of edges.

• ​​ (Tensor[Float, [Bs3]]) – Distance of x to the closest projection.

• ​​​ (Tensor[Float, [Bs3]]) – Index of the edge that was projected on.