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.

__init__(rotation: Rot3Array, translation: Vec3Array) None

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

device

dtype

shape

rotation

translation