src/transformers/models/esm/openfold_utils/rigid_utils.py/Rigid/shape
class Rigid:    def shape(self) -> torch.Size:
        """
        Returns the shape of the shared dimensions of the rotation and the translation.

        Returns:
            The shape of the transformation
        """
        return self._trans.shape[:-1]