miniworld.utils.kinematics.xyz_to_c6d¶

miniworld.utils.kinematics.xyz_to_c6d(xyz, params={'ABINS': 36, 'DBINS': 36, 'DMAX': 20.0, 'DMIN': 2.0})[source]¶

convert cartesian coordinates into 2d distance and orientation maps

Parameters:: xyz (pytorch tensor of shape [batch,nres,3,3]) – stores Cartesian coordinates of backbone N,Ca,C atoms
Returns:: c6d – stores stacked dist,omega,theta,phi 2D maps
Return type:: pytorch tensor of shape [batch,nres,nres,4]