deepfold.utils.geometry.rigid_matrix_vector.Rigid3Array¶
- class deepfold.utils.geometry.rigid_matrix_vector.Rigid3Array(rotation: Rot3Array, translation: Vec3Array)[source]¶
Rigid Transformation, i.e. element of special euclidean group.
Methods
__init__(rotation, translation)apply(point)apply_inverse_to_point(point)Apply inverse Rigid3Array transform to point.
apply_to_point(point)Apply Rigid3Array transform to point.
cat(rigids, dim)compose(other_rigid)compose_rotation(other_rotation)cuda()from_array(array)from_array4x4(array)Construct Rigid3Array from homogeneous 4x4 array.
from_tensor_4x4(array)identity(shape, device)Return identity Rigid3Array of given shape.
inverse()Return Rigid3Array corresponding to inverse transform.
invert_apply(point)map_tensor_fn(fn)reshape(new_shape)scale_translation(factor)Scale translation in Rigid3Array by 'factor'.
stop_rot_gradient()to_tensor()to_tensor_4x4()unsqueeze(dim)Attributes
devicedtypeshaperotationtranslation