If we have a set of points Rs
, we can use torch.cdist to get the all pair distances.
dists_ij = torch.cdist(Rs, Rs)
Is there a function to get the angles between two set of vectors Vs
like this:
angs_ij = torch.angs(Vs, Vs)
from Calculate angles in pytorch
No comments:
Post a Comment