diff --git a/src/mrpro/utils/reshape.py b/src/mrpro/utils/reshape.py index 742d4f55e..2a6923337 100644 --- a/src/mrpro/utils/reshape.py +++ b/src/mrpro/utils/reshape.py @@ -100,7 +100,7 @@ def reduce_view(x: torch.Tensor, dim: int | Sequence[int] | None = None) -> torc ] return torch.as_strided(x, newsize, stride) - +@lru_cache def _reshape_idx(old_shape: tuple[int, ...], new_shape: tuple[int, ...], old_stride: tuple[int, ...]) -> list[slice]: """Get reshape reduce index (Cached helper function for reshape_broadcasted)