# Import from the canonical location in torch.func
from torch._functorch.einops import rearrange


__all__ = ["rearrange"]