#StackBounty: #python #pytorch Calculate angles in pytorch

Bounty: 50

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)


Get this bounty!!!

Leave a Reply

This site uses Akismet to reduce spam. Learn how your comment data is processed.