miniworld.utils.kinematics.get_ang

miniworld.utils.kinematics.get_ang(a, b, c)[source]

calculate planar angles for all consecutive triples (a[i],b[i],c[i]) from Cartesian coordinates of three sets of atoms a,b,c

Parameters:

  • a (pytorch tensors of shape [batch,nres,3]) – store Cartesian coordinates of three sets of atoms

  • b (pytorch tensors of shape [batch,nres,3]) – store Cartesian coordinates of three sets of atoms

  • c (pytorch tensors of shape [batch,nres,3]) – store Cartesian coordinates of three sets of atoms

Returns:

ang – stores resulting planar angles

Return type:

pytorch tensor of shape [batch,nres]