Method apply

modelscope/models/science/unifold/modules/frame.py:279–284

(
        self,
        pts: torch.Tensor,
    )

        return Frame(new_rot, new_trans)

    def apply(
        self,
        pts: torch.Tensor,
    ) -> torch.Tensor:
        rotated = self._r.apply(pts)
        return rotated + self._t

    def invert_apply(self, pts: torch.Tensor) -> torch.Tensor:
        pts = pts - self._t