Module util

Module util 

Source
Expand description

Shared helpers (permutation inversion, multi-index iteration, dimension fusion). Shared helpers for strided-einsum2.

Structs§

MultiIndex
Iterator over multi-dimensional index tuples within given dimensions.

Functions§

invert_perm
Invert a permutation: if perm[i] = j, then result[j] = i.
try_fuse_group
Try to fuse a contiguous dimension group into a single (total_size, innermost_stride).