Method __init__

modelscope/models/science/unifold/modules/frame.py:185–212

(
        self,
        rotation: Optional[Rotation],
        translation: Optional[torch.Tensor],
    )

class Frame:

    def __init__(
        self,
        rotation: Optional[Rotation],
        translation: Optional[torch.Tensor],
    ):
        if rotation is None and translation is None:
            rotation = Rotation.identity((0, ))
            translation = zero_translation((0, ))
        elif translation is None:
            translation = zero_translation(rotation.shape, rotation.dtype,
                                           rotation.device,
                                           rotation.requires_grad)

        elif rotation is None:
            rotation = Rotation.identity(
                translation.shape[:-1],
                translation.dtype,
                translation.device,
                translation.requires_grad,
            )

        if (rotation.shape != translation.shape[:-1]) or (rotation.device
                                                          !=  # noqa W504
                                                          translation.device):
            raise ValueError('RotationMatrix and translation incompatible')

        self._r = rotation
        self._t = translation

    @staticmethod
    def identity(