tfep.utils.geometry.vector_plane_angle

tfep.utils.geometry.vector_plane_angle(x, plane)

Return the angle in radians between a batch of vectors and another vector.

Parameters:

• x (torch.Tensor) – A tensor of shape (batch_size, N) or (N,).

• plane (torch.Tensor) – A tensor of shape (N,) that represent a normal vector to the plane.

Returns:

angle – A tensor of shape (batch_size,) where angle[i] is the angle between vector x[i] and plane plane.

Return type:

torch.Tensor