Tuesday 20 July 2021

Calculate angles in pytorch

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