Skip to content

Commit

Permalink
enable cache
Browse files Browse the repository at this point in the history
  • Loading branch information
fzimmermann89 authored Dec 2, 2024
1 parent 6c97ee5 commit b330f20
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/mrpro/utils/reshape.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit b330f20

Please sign in to comment.