miniworld.utils.kinematics.get_pair_dist

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

calculate pair distances between two sets of points

Parameters:

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

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

Returns:

dist – stores paitwise distances between atoms in a and b

Return type:

pytorch tensor of shape [batch,nres,nres]