einops-exts icon indicating copy to clipboard operation
einops-exts copied to clipboard

Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️

Results 1 einops-exts issues
Sort by recently updated
recently updated
newest added

Using `rearrange_many` in a model results in a very nondescript error when running `torch.compile(model)` However, replacing `rearrange_many` with `rearrange` fixes it. Could it be related to your use of wrapper/higher-order...