deepfold.utils.geometry.vector.square_euclidean_distance¶

deepfold.utils.geometry.vector.square_euclidean_distance(vec1: Vec3Array, vec2: Vec3Array, epsilon: float = 1e-06) → float | Tensor[source]¶

Computes square of euclidean distance between ‘vec1’ and ‘vec2’.

Parameters:

vec1 – Vec3Array to compute distance to
vec2 – Vec3Array to compute distance from, should be broadcast compatible with ‘vec1’
epsilon – distance is clipped from below to be at least epsilon

Returns:

Array of square euclidean distances; shape will be result of broadcasting ‘vec1’ and ‘vec2’