File size: 123 Bytes
a40a8f5 | 1 2 3 4 5 6 | # Import from the canonical location in torch.func
from torch._functorch.einops import rearrange
__all__ = ["rearrange"]
|
a40a8f5 | 1 2 3 4 5 6 | # Import from the canonical location in torch.func
from torch._functorch.einops import rearrange
__all__ = ["rearrange"]
|